fixture shell to remove zoutpu

marrlab · Dec 11, 2024 · 2a9f862 · 2a9f862
1 parent 39f9a0d
commit 2a9f862
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -22,7 +22,7 @@ jobs:
       run: |
         python -m pip install --upgrade pip
         pip install -r requirements.txt
-        pip install pytest pytest-cov pytest-shell
+        pip install pytest pytest-cov pytest-shell pytest-shell-utilities
         python setup.py install
     - name: Install Python Poetry
       run: |

diff --git a/tests/utils_test.py b/tests/utils_test.py
@@ -22,6 +22,8 @@ def utils_test_algo(argsstr="--help"):
     del exp
     torch.cuda.empty_cache()
     gc.collect()
+    ret = shell.run("rm", "-rf zoutput")
+    assert ret.returncode == 0
 
 
 def assert_frame_not_equal(*args, **kwargs):