Update displacements tests plots

SMTG-Bham · Dec 7, 2024 · f5a32a2 · f5a32a2
1 parent fa928da
commit f5a32a2
Show file tree

Hide file tree

Showing 12 changed files with 14 additions and 0 deletions.
diff --git a/tests/data/remote_baseline_plots/F_i_-1_disp_anisotropy_plot.png b/tests/data/remote_baseline_plots/F_i_-1_disp_anisotropy_plot.png
diff --git a/tests/data/remote_baseline_plots/F_i_-1_disp_ellipsoid_plot.png b/tests/data/remote_baseline_plots/F_i_-1_disp_ellipsoid_plot.png
diff --git a/tests/data/remote_baseline_plots/Te_Cd_1_disp_anisotropy_plot.png b/tests/data/remote_baseline_plots/Te_Cd_1_disp_anisotropy_plot.png
diff --git a/tests/data/remote_baseline_plots/Te_Cd_1_disp_ellipsoid_plot.png b/tests/data/remote_baseline_plots/Te_Cd_1_disp_ellipsoid_plot.png
diff --git a/tests/data/remote_baseline_plots/YTOS_Int_F_-1_site_displacements.png b/tests/data/remote_baseline_plots/YTOS_Int_F_-1_site_displacements.png
diff --git a/tests/data/remote_baseline_plots/v_Cd_0_disp_anisotropy_plot.png b/tests/data/remote_baseline_plots/v_Cd_0_disp_anisotropy_plot.png
diff --git a/tests/data/remote_baseline_plots/v_Cd_0_disp_ellipsoid_plot.png b/tests/data/remote_baseline_plots/v_Cd_0_disp_ellipsoid_plot.png
diff --git a/tests/data/remote_baseline_plots/v_Cd_0_disp_plot.png b/tests/data/remote_baseline_plots/v_Cd_0_disp_plot.png
diff --git a/tests/data/remote_baseline_plots/v_Cd_0_disp_plot_relaxed_dists.png b/tests/data/remote_baseline_plots/v_Cd_0_disp_plot_relaxed_dists.png
diff --git a/tests/data/remote_baseline_plots/v_Cd_0_disp_proj_plot.png b/tests/data/remote_baseline_plots/v_Cd_0_disp_proj_plot.png
diff --git a/tests/data/remote_baseline_plots/v_Cd_0_disp_proj_plot_relaxed_dists.png b/tests/data/remote_baseline_plots/v_Cd_0_disp_proj_plot_relaxed_dists.png
diff --git a/tests/test_displacements.py b/tests/test_displacements.py
@@ -260,6 +260,20 @@ def test_plot_site_displacements(self):
             separated_by_direction=False, use_plotly=False
         )
 
+    @custom_mpl_image_compare(filename="v_Cd_0_disp_proj_plot_relaxed_dists.png")
+    def test_plot_site_displacements_proj_relaxed_dists(self):
+        # Vacancy, displacement separated by direction:
+        return self.v_Cd_0_defect_entry.plot_site_displacements(
+            separated_by_direction=True, use_plotly=False, relaxed_distances=True
+        )
+
+    @custom_mpl_image_compare(filename="v_Cd_0_disp_plot_relaxed_dists.png")
+    def test_plot_site_displacements_relaxed_dists(self):
+        # Vacancy, total displacement
+        return self.v_Cd_0_defect_entry.plot_site_displacements(
+            separated_by_direction=False, use_plotly=False, relaxed_distances=True
+        )
+
     @custom_mpl_image_compare(filename="YTOS_Int_F_-1_site_displacements.png")
     def test_plot_site_displacements_ytos(self):
         # Interstitial, total displacement