modify test_displacement

SMTG-Bham · Sep 19, 2024 · 86bbf54 · 86bbf54
1 parent 29be195
commit 86bbf54
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_displacements.py b/tests/test_displacements.py
@@ -210,14 +210,14 @@ def test_plot_site_displacements_ytos(self):
 
     @custom_mpl_image_compare(filename="v_Cd_0_disp_ellipsoid_plot.png")
     def test_calc_displacements_ellipsoid_ellipsoid(self):
-        defect_entry = core.DefectEntry.from_json(f"{data_dir}/v_Cd_defect_entry.json.gz")
+        defect_entry = DefectEntry.from_json(f"{data_dir}/v_Cd_defect_entry.json.gz")
         return_tuple = calc_displacements_ellipsoid(defect_entry, plot_ellipsoid=True)
         # return fig object
         return return_tuple[3]
 
     @custom_mpl_image_compare(filename="v_Cd_0_disp_anisotropy_plot.png")
     def test_calc_displacements_ellipsoid_anisotropy(self):
-        defect_entry = core.DefectEntry.from_json(f"{data_dir}/v_Cd_defect_entry.json.gz")
+        defect_entry = DefectEntry.from_json(f"{data_dir}/v_Cd_defect_entry.json.gz")
         return_tuple = calc_displacements_ellipsoid(defect_entry, plot_anisotropy=True)
         # return fig object
         return return_tuple[3]