Fixed syntax errors and broken tests.

spc-group · Dec 2, 2022 · b8a8159 · b8a8159
1 parent 55121f6
commit b8a8159
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/haven/plans/mono_gap_calibration.py b/haven/plans/mono_gap_calibration.py
@@ -12,6 +12,7 @@
 from ..instrument.instrument_registry import registry
 from .set_energy import set_energy
 from .auto_gain import auto_gain
+from .beam_properties import knife_scan
 from .._iconfig import load_config
 
 

diff --git a/tests/test_beam_properties.py b/tests/test_beam_properties.py
@@ -143,7 +143,7 @@ def test_fit_step():
     )
     expected_center = 5960.94  # 5961.21
     # Do the fitting
-    result = fit_step(x, y, plot=True, plot_derivative=True)
+    result = fit_step(x, y, plot=False, plot_derivative=False)
     plt.show()
     assert result.position == pytest.approx(expected_center)
     assert result.fwhm == pytest.approx(1449.956)

diff --git a/tests/test_monochromator.py b/tests/test_monochromator.py
@@ -12,9 +12,7 @@ def test_mono_energy_signal(ioc_mono):
                          energy_prefix="mono_ioc",
                          name="monochromator")
     time.sleep(0.1)
-    mono.wait_for_connection(timeout=20)
     # Change mono energy
     mono.energy.set(5000)
-    time.sleep(1.)
     # Check new value on the IOC
-    assert epics.caget("mono_ioc:Energy") == 5000
+    assert epics.caget("mono_ioc:Energy", use_monitor=False) == 5000