clean up test

Open-EO/openeo-geopyspark-driver#883

tests/test_workspace.py

@@ -1,5 +1,3 @@
-import os
-
 import pytest
 
 from openeo_driver.workspace import DiskWorkspace
@@ -24,8 +22,8 @@ def test_disk_workspace(tmp_path, merge):
     workspace = DiskWorkspace(root_directory=tmp_path)
     workspace.import_file(file=source_file, merge=merge)
 
-    assert source_file.name in os.listdir(target_directory)
-    assert source_file.name in os.listdir(source_directory)
+    assert (target_directory / source_file.name).exists()
+    assert (source_directory / source_file.name).exists()
 
 
 @pytest.mark.parametrize("remove_original", [False, True])
@@ -41,5 +39,5 @@ def test_disk_workspace_remove_original(tmp_path, remove_original):
     workspace = DiskWorkspace(root_directory=tmp_path)
     workspace.import_file(source_file, merge=merge, remove_original=remove_original)
 
-    assert source_file.name in os.listdir(target_directory)
-    assert remove_original == (source_file.name not in os.listdir(source_directory))
+    assert (target_directory / source_file.name).exists()
+    assert (source_directory / source_file.name).exists() != remove_original