Skip to content

Commit 37fdb7c

Browse files
committed
Added unit test
1 parent 5e8018b commit 37fdb7c

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

tests/cut_plotter_presenter_test.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -334,6 +334,17 @@ def test_propagate_sample_temperatures_throughout_cache(self):
334334
self.assertEqual(cut_2.sample_temp, 120)
335335
self.assertEqual(cut_3._sample_temp, None)
336336

337+
def test_missing_sample_temperature(self):
338+
ax = mock.MagicMock()
339+
self.populate_presenter_cache_dict(ax)
340+
cut_1, cut_2 = self.cut_plotter_presenter._cut_cache_dict[ax]
341+
cut_1.sample_temp = 120
342+
self.cut_plotter_presenter._get_overall_max_signal(
343+
IntensityType.SCATTERING_FUNCTION
344+
)
345+
self.assertEqual(cut_1.sample_temp, 120)
346+
self.assertEqual(cut_2.sample_temp, 120)
347+
337348
@mock.patch("mslice.presenters.cut_plotter_presenter.plt.gca")
338349
def test_get_overall_q_axis(self, mock_plot_gca):
339350
ax = mock.MagicMock()

0 commit comments

Comments
 (0)