tests: skip append parquet tests if fastparquet is not installed

Avaiga · Nov 13, 2023 · 2682eb4 · 2682eb4
1 parent 5812d79
commit 2682eb4
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/tests/core/data/test_parquet_data_node.py b/tests/core/data/test_parquet_data_node.py
@@ -300,6 +300,7 @@ def test_get_system_folder_modified_date_instead_of_last_edit_date(self, tmpdir_
 
         os.unlink(temp_file_path)
 
+    @pytest.mark.skipif(not util.find_spec("fastparquet"), reason="Append parquet requires fastparquet to be installed")
     @pytest.mark.parametrize(
         "content",
         [
@@ -308,10 +309,6 @@ def test_get_system_folder_modified_date_instead_of_last_edit_date(self, tmpdir_
         ],
     )
     def test_append_pandas(self, parquet_file_path, default_data_frame, content):
-        # !!! append data only works with `fastparquet`
-        if not util.find_spec("fastparquet"):
-            pass
-
         dn = ParquetDataNode("foo", Scope.SCENARIO, properties={"path": parquet_file_path})
         assert_frame_equal(dn.read(), default_data_frame)
 
@@ -321,6 +318,7 @@ def test_append_pandas(self, parquet_file_path, default_data_frame, content):
             pd.concat([default_data_frame, pd.DataFrame(content, columns=["a", "b", "c"])]).reset_index(drop=True),
         )
 
+    @pytest.mark.skipif(not util.find_spec("fastparquet"), reason="Append parquet requires fastparquet to be installed")
     @pytest.mark.parametrize(
         "content",
         [
@@ -329,10 +327,6 @@ def test_append_pandas(self, parquet_file_path, default_data_frame, content):
         ],
     )
     def test_append_modin(self, parquet_file_path, default_data_frame, content):
-        # !!! append data only works with `fastparquet`
-        if not util.find_spec("fastparquet"):
-            pass
-
         dn = ParquetDataNode("foo", Scope.SCENARIO, properties={"path": parquet_file_path, "exposed_type": "modin"})
         df_equals(dn.read(), modin_pd.DataFrame(default_data_frame))