From 86808072fa9a92ce978aebaba01c51374ab8dbf2 Mon Sep 17 00:00:00 2001 From: Sandro Loch Date: Tue, 5 Dec 2023 11:57:53 -0300 Subject: [PATCH] Reduce the size of the df for tests --- pysus/tests/test_data/test_Infodengue.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/pysus/tests/test_data/test_Infodengue.py b/pysus/tests/test_data/test_Infodengue.py index 82aef58f..c80db9cc 100644 --- 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,7 +87,7 @@ def test_normalize(self): def test_download(self): df = download( "dengue", - 202101, + 202129, 202152, "Rio de Janeiro", )