remove other files generated by unit tests

virtualcell · Aug 15, 2024 · 8b41c28 · 8b41c28
1 parent 7f7b037
commit 8b41c28
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/fvsolver_test.py b/tests/fvsolver_test.py
@@ -27,4 +27,10 @@ def test_solve() -> None:
     assert test_output_dir_1.exists()
     assert len(list(test_output_dir_1.iterdir())) > 0
 
+    # empty directory test_output_dir_1 and test_output_dir_2
+    for file in test_output_dir_1.iterdir() if test_output_dir_1.exists() else []:
+        file.unlink()
+    for file in test_output_dir_2.iterdir() if test_output_dir_2.exists() else []:
+        file.unlink()
+
     print(f"retcode_1: {retcode_1}")