Fix runaway test teardown method

SMTG-Bham · Apr 6, 2024 · ef6df93 · ef6df93
1 parent a68ecfa
commit ef6df93
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_vasp.py b/tests/test_vasp.py
@@ -593,7 +593,10 @@ def tearDown(self):
  if_present_rm("CdTe_bulk")
 
  for i in os.listdir():
- if os.path.isdir(i) and ("MgO" in i or any("vasp" in j for j in os.listdir(i))):
+ if os.path.isdir(i) and (
+ "MgO" in i
+ or any("vasp_" in j and os.path.isdir(os.path.join(i, j)) for j in os.listdir(i))
+ ):
  if_present_rm(i)
 
  if_present_rm("MgO_defects_generator.json")