Update fastparquet version

AlertaDengue · Dec 12, 2023 · 14a0fe8 · 14a0fe8
1 parent e7d3615
commit 14a0fe8
Show file tree

Hide file tree

Showing 6 changed files with 65 additions and 53 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,7 +12,7 @@ python = ">=3.10,<3.12"
 python-dateutil = "2.8.2"
 cffi = "1.15.1"
 dbfread = "2.0.7"
-fastparquet = "^0.8.1"
+fastparquet = ">=2023.10.1"
 numpy = "1.26.2"
 pyarrow = ">=11.0.0"
 pycparser = "2.21"

diff --git a/pysus/tests/test_decoders.py b/pysus/tests/test_decoders.py
@@ -71,7 +71,7 @@ def test_decodifica_idade_retorna_em_anos_SIM(self):
     def test_verifica_geocodigo(self):
         self.assertTrue(decoders.is_valid_geocode(3304557))
 
-    @pytest.mark.timeout(10)
+    @pytest.mark.timeout(60)
     def test_translate_variables(self):
         df = download(groups="cid10", states="sp", years=2010).to_dataframe()
         df = decoders.translate_variables_SIM(df)
@@ -81,7 +81,7 @@ def test_translate_variables(self):
         assert raca_array <= set(
             ["Branca", "Preta", "Amarela", "nan", "Parda", "Indígena", "NA"])
 
-    @pytest.mark.timeout(10)
+    @pytest.mark.timeout(60)
     def test_get_cid_chapter(self):
         code_index = decoders.get_CID10_code_index(get_CID10_chapters_table())
         test_causes = pd.DataFrame(
@@ -109,7 +109,7 @@ def test_get_cid_chapter(self):
         assert_array_equal(
             results, [1, 1, 2, -1, 3, 7, 7, 8, -1, 20, 20, -1, 22])
 
-    @pytest.mark.timeout(10)
+    @pytest.mark.timeout(60)
     def test_group_and_count(self):
         df = download(groups="cid10", states="se", years=2010).to_dataframe()
         df = decoders.translate_variables_SIM(df)
@@ -121,7 +121,7 @@ def test_group_and_count(self):
         )
         self.assertGreater(sum(sample), 0)
 
-    @pytest.mark.timeout(10)
+    @pytest.mark.timeout(60)
     def test_redistribute(self):
         df = download(groups="cid10", states="sp", years=2010).to_dataframe()
         df = decoders.translate_variables_SIM(

diff --git a/pysus/tests/test_ftp/test_Directory.py b/pysus/tests/test_ftp/test_Directory.py
diff --git a/pysus/tests/test_ftp/test_File.py b/pysus/tests/test_ftp/test_File.py
diff --git a/pysus/tests/test_ftp/test_databases/test_SIA.py b/pysus/tests/test_ftp/test_databases/test_SIA.py
@@ -47,8 +47,8 @@ def test_sia(self):
         }
 
         with patch(
-                'pysus.ftp.databases.sia.SIA',
-                return_value=MagicMock(__content__=mock_content)
+            'pysus.ftp.databases.sia.SIA',
+            return_value=MagicMock(__content__=mock_content)
         ) as mock_sia:
             sia = SIA()
             sia.__content__ = mock_sia().__content__
@@ -63,19 +63,19 @@ def test_sia(self):
                  'size': '3.0 kB',
                  'last_update': '2019-03-12 12:03PM'},
                 {'name': 'ABMG1112.dbc',
-                    'group': 'APAC de Cirurgia Bariátrica',
-                    'uf': 'Minas Gerais',
-                    'month': 'Dezembro',
-                    'year': 2011,
-                    'size': '3.2 kB',
-                    'last_update': '2019-03-12 12:03PM'},
+                 'group': 'APAC de Cirurgia Bariátrica',
+                 'uf': 'Minas Gerais',
+                 'month': 'Dezembro',
+                 'year': 2011,
+                 'size': '3.2 kB',
+                 'last_update': '2019-03-12 12:03PM'},
                 {'name': 'ABOAC1502.dbc',
-                    'group': 'APAC de Acompanhamento Pós Cirurgia Bariátrica',
-                    'uf': 'Acre',
-                    'month': 'Fevereiro',
-                    'year': 2015,
-                    'size': '3.1 kB',
-                    'last_update': '2016-09-12 08:45AM'}
+                 'group': 'APAC de Acompanhamento Pós Cirurgia Bariátrica',
+                 'uf': 'Acre',
+                 'month': 'Fevereiro',
+                 'year': 2015,
+                 'size': '3.1 kB',
+                 'last_update': '2016-09-12 08:45AM'}
             ]
 
             self.assertEqual(descriptions, expected_descriptions)