Browse Source

mark modin test

jrobinAV 1 year ago
parent
commit
84c279de20
1 changed files with 15 additions and 0 deletions
  1. 15 0
      tests/core/data/test_parquet_data_node.py

+ 15 - 0
tests/core/data/test_parquet_data_node.py

@@ -358,6 +358,21 @@ class TestParquetDataNode:
         assert pathlib.Path(temp_file_path).exists()
         assert isinstance(dn.read(), pd.DataFrame)
 
+        @pytest.mark.modin
+        @pytest.mark.parametrize(
+            "data",
+            [
+                modin_pd.DataFrame([{"a": 11, "b": 22, "c": 33}, {"a": 44, "b": 55, "c": 66}]),
+            ],
+        )
+        def test_write_to_disk_modin(self, tmpdir_factory, data):
+            temp_file_path = str(tmpdir_factory.mktemp("data").join("temp.parquet"))
+            dn = ParquetDataNode("foo", Scope.SCENARIO, properties={"path": temp_file_path})
+            dn.write(data)
+
+            assert pathlib.Path(temp_file_path).exists()
+            assert isinstance(dn.read(), pd.DataFrame)
+
     @pytest.mark.modin
     @pytest.mark.parametrize(
         "data",