Reduce the size of the df for tests

AlertaDengue · Dec 5, 2023 · 6b29ce2 · 6b29ce2
1 parent fdac834
commit 6b29ce2
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/pysus/tests/test_data/test_Infodengue.py b/pysus/tests/test_data/test_Infodengue.py
@@ -1,12 +1,8 @@
 import unittest
-import pytest
 
 import pandas as pd
-from pysus.online_data.Infodengue import (
-    download,
-    search_string,
-    normalize,
-)
+import pytest
+from pysus.online_data.Infodengue import download, normalize, search_string
 
 
 class InfoDengueTestCase(unittest.TestCase):
@@ -91,11 +87,11 @@ def test_normalize(self):
     def test_download(self):
         df = download(
             "dengue",
-            202101,
+            202129,
             202152,
             "Rio de Janeiro",
         )
-        df_size = (29, 52)
+        df_size = (29, 24)
 
         self.assertIsInstance(df, pd.DataFrame)
         self.assertGreater(len(df), 0)