Skip to content

Commit 2d372a5

Browse files
committed
get cor and tilt from view and added in expected values
1 parent 548e0ea commit 2d372a5

File tree

1 file changed

+22
-15
lines changed

1 file changed

+22
-15
lines changed

mantidimaging/gui/test/gui_system_reconstruction_test.py

Lines changed: 22 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -44,28 +44,34 @@ def tearDown(self) -> None:
4444
self.assertFalse(self.main_window.isVisible())
4545

4646
def test_correlate(self):
47-
expected_initial_cor = self.recon_window.presenter.model.last_cor.value
48-
expected_initial_tilt = self.recon_window.presenter.model.auto_find_correlation(progress=None)[1].value
47+
expected_initial_cor = self.recon_window.rotation_centre
48+
expected_initial_tilt = self.recon_window.tilt
49+
50+
expected_final_cor = 66.0
51+
expected_final_tilt = 1.070368
4952

5053
for _ in range(5):
5154
QTest.mouseClick(self.recon_window.correlateBtn, Qt.MouseButton.LeftButton)
5255
QTest.qWait(SHORT_DELAY)
53-
5456
wait_until(lambda: self.recon_window.correlateBtn.isEnabled())
5557
wait_until(lambda: len(self.recon_window.presenter.async_tracker) == 0)
5658

57-
final_cor_value = self.recon_window.presenter.model.last_cor.value
58-
final_tilt_value = self.recon_window.presenter.model.auto_find_correlation(progress=None)[1].value
59+
final_cor_value = self.recon_window.rotation_centre
60+
final_tilt_value = self.recon_window.tilt
5961

60-
if final_cor_value != expected_initial_cor:
61-
assert final_cor_value != expected_initial_cor
62+
assert final_cor_value != expected_initial_cor
63+
assert final_tilt_value != expected_initial_tilt
6264

63-
assert final_tilt_value == expected_initial_tilt
65+
assert final_cor_value == expected_final_cor
66+
assert final_tilt_value == expected_final_tilt
6467

6568
@pytest.mark.xfail(reason="Unresolved, see #1641")
6669
def test_minimise(self):
67-
expected_initial_cor = self.recon_window.presenter.model.last_cor.value
68-
expected_initial_tilt = self.recon_window.presenter.model.auto_find_correlation(progress=None)[1].value
70+
expected_cor = 64.0
71+
expected_tilt = 0.0
72+
73+
initial_cor = self.recon_window.rotation_centre
74+
initial_tilt = self.recon_window.tilt
6975

7076
for i in range(2, 6):
7177
QTimer.singleShot(SHORT_DELAY, lambda i=i: self._click_InputDialog(set_int=i))
@@ -76,13 +82,14 @@ def test_minimise(self):
7682
wait_until(lambda: len(self.recon_window.presenter.async_tracker) == 0)
7783
QTest.qWait(SHORT_DELAY)
7884

79-
final_cor_value = self.recon_window.presenter.model.last_cor.value
80-
final_tilt_value = self.recon_window.presenter.model.auto_find_correlation(progress=None)[1].value
85+
final_cor_value = self.recon_window.rotation_centre
86+
final_tilt_value = self.recon_window.tilt
8187

82-
if final_cor_value != expected_initial_cor:
83-
assert final_cor_value != expected_initial_cor
88+
assert initial_cor == expected_cor
89+
assert initial_tilt == expected_tilt
8490

85-
assert final_tilt_value == expected_initial_tilt
91+
assert final_cor_value == expected_cor
92+
assert final_tilt_value == expected_tilt
8693

8794
@classmethod
8895
def _click_cor_inspect(cls):

0 commit comments

Comments
 (0)