Changed test_dataset_extract_remove_finished_false to include new param

aeye-lab · Sep 27, 2023 · 2c643dd · 2c643dd
1 parent 22bf1f8
commit 2c643dd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/dataset/dataset_download_test.py b/tests/dataset/dataset_download_test.py
@@ -214,14 +214,15 @@ def test_dataset_extract_remove_finished_true(
 
     paths = pm.DatasetPaths(root=tmp_path, dataset='.')
     dataset = pm.Dataset(dataset_definition, path=paths)
-    dataset.extract(remove_finished=True, verbose=1)
+    dataset.extract(remove_finished=True, remove_top_level=False, verbose=1)
 
     mock_extract_archive.assert_has_calls([
         mock.call(
             source_path=tmp_path / 'downloads' / 'test.gz.tar',
             destination_path=tmp_path / 'raw',
             recursive=True,
             remove_finished=True,
+            remove_top_level=False,
             verbose=1,
         ),
     ])