add make_fragment to test

apache · May 3, 2022 · a0e0ad8 · a0e0ad8
1 parent cd9b58d
commit a0e0ad8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/pyarrow/tests/test_dataset.py b/python/pyarrow/tests/test_dataset.py
@@ -2602,10 +2602,9 @@ def test_open_dataset_from_fsspec(tempdir):
 
 @pytest.mark.parquet
 @pytest.mark.s3
-def test_inspect_file_fsspec(s3_filesystem):
+def test_file_format_inspect_fsspec(s3_filesystem):
     # https://issues.apache.org/jira/browse/ARROW-16413
     from pyarrow.fs import _ensure_filesystem
-    import pyarrow.dataset as ds
 
     fs, (host, port, access_key, secret_key) = s3_filesystem
 
@@ -2629,6 +2628,9 @@ def test_inspect_file_fsspec(s3_filesystem):
     schema = format.inspect(path, filesystem)
     assert schema.equals(table.schema)
 
+    fragment = format.make_fragment(path, filesystem)
+    assert fragment.physical_schema.equals(table.schema)
+
 
 @pytest.mark.pandas
 def test_filter_timestamp(tempdir, dataset_reader):