Skip to content

Commit 1b4219c

Browse files
committed
Updated test_beamcut_mds.py to also test for the phase plot.
1 parent f371a00 commit 1b4219c

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tests/unit/mdses/test_beamcut_mds.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ def test_beamcut_mds_plots(self):
6666
amp_plot_name = f"beamcut_amplitude_ant_{ant}_ddi_{ddi}.png"
6767
att_plot_name = f"beamcut_attenuation_ant_{ant}_ddi_{ddi}.png"
6868
lm_plot_name = f"beamcut_lm_offsets_ant_{ant}_ddi_{ddi}.png"
69+
pha_plot_name = f"beamcut_phase_ant_{ant}_ddi_{ddi}.png"
6970

7071
beamcut_mds = open_beamcut(self.remote_beamcut_name)
7172

@@ -96,6 +97,14 @@ def test_beamcut_mds_plots(self):
9697
)
9798
assert equal, f"{msg}: lm plot png file is different from the expected png file"
9899

100+
beamcut_mds.plot_beamcut_in_phase(self.destination_folder, ant=ant, ddi=ddi)
101+
equal, msg = are_png_files_close(
102+
f"{self.destination_folder}/{pha_plot_name}",
103+
f"{self.ref_products_name}/{pha_plot_name}",
104+
)
105+
assert (
106+
equal
107+
), f"{msg}: phase plot png file is different from the expected png file"
99108
return
100109

101110
def test_beamcut_mds_fit_report(self):

0 commit comments

Comments
 (0)