Update teardown methods

SMTG-Bham · Sep 9, 2024 · 2ef5ee1 · 2ef5ee1
1 parent 31795f3
commit 2ef5ee1
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/tests/test_analysis.py b/tests/test_analysis.py
@@ -2511,6 +2511,9 @@ def tearDown(self):
  if_present_rm(os.path.join(self.Cu2SiSe3_EXAMPLE_DIR, "Cu2SiSe3_defect_dict.json.gz"))
  if_present_rm(os.path.join(self.Cu2SiSe3_EXAMPLE_DIR, "bulk/vasp_std/voronoi_nodes.json"))
 
+ if_present_rm(os.path.join(self.ZnS_DATA_DIR, "ZnS_defect_dict.json.gz"))
+ if_present_rm(os.path.join(self.ZnS_DATA_DIR, "bulk/voronoi_nodes.json"))
+
  def test_defect_name_from_structures(self):
  # by proxy also tests defect_from_structures
  for defect_gen_name in [
@@ -3362,6 +3365,10 @@ def setUp(self):
  self.CdTe_corrections_dir = os.path.join(self.module_path, "data/CdTe_charge_correction_tests")
  self.v_Cd_m2_path = f"{self.CdTe_corrections_dir}/v_Cd_-2_vasp_gam"
  self.CdTe_dielectric = np.array([[9.13, 0, 0], [0.0, 9.13, 0], [0, 0, 9.13]]) # CdTe
+ self.CdTe_BULK_DATA_DIR = os.path.join(self.CdTe_EXAMPLE_DIR, "CdTe_bulk/vasp_ncl")
+
+ def tearDown(self):
+ if_present_rm(os.path.join(self.CdTe_BULK_DATA_DIR, "voronoi_nodes.json"))
 
  @custom_mpl_image_compare(filename="CdTe_v_cd_m2_eigenvalue_plot.png")
  def test_parsing_cdte(self):

diff --git a/tests/test_vasp.py b/tests/test_vasp.py
@@ -1133,6 +1133,8 @@ def tearDown(self):
  if_present_rm(folder)
 
  if_present_rm("AgSbTe2_test")
+ if_present_rm("CdTe_defects_generator.json")
+ if_present_rm("test_CdTe_defects_generator.json")
 
  def check_generated_vasp_inputs(
  self,