def test_stirring_with_lookup_linear_v1() -> None:
class FakeRpmCalculator:
def __call__(self, *args):
return 475
def cleanup(self):
pass
with local_persistant_storage("stirring_calibration") as cache:
cache["linear_v1"] = json.dumps({"rpm_coef": 0.1, "intercept": 20})
target_rpm = 500
current_dc = Stirrer.duty_cycle
with Stirrer(target_rpm, unit, exp,
rpm_calculator=FakeRpmCalculator()) as st: # type: ignore
st.start_stirring()
assert st.duty_cycle == current_dc - 0.9 * (current_dc -
(0.1 * target_rpm + 20))
pause()
pause()
current_dc = st.duty_cycle
target_rpm = 600
publish(f"pioreactor/{unit}/{exp}/stirring/target_rpm/set", target_rpm)
pause()
pause()
assert st.duty_cycle == current_dc - 0.9 * (current_dc -
(0.1 * target_rpm + 20))
def test_publish_duty_cycle():
publish(f"pioreactor/{unit}/{exp}/stirring/duty_cycle", None, retain=True)
pause()
original_dc = 50
st = Stirrer(original_dc, unit, exp)
assert st.duty_cycle == original_dc
pause()
message = subscribe(f"pioreactor/{unit}/{exp}/stirring/duty_cycle")
assert float(message.payload) == 50
def test_publish_target_rpm() -> None:
publish(f"pioreactor/{unit}/{exp}/stirring/target_rpm", None, retain=True)
pause()
target_rpm = 500
with Stirrer(target_rpm, unit, exp, rpm_calculator=RpmCalculator()) as st:
assert st.target_rpm == target_rpm
pause()
message = subscribe(f"pioreactor/{unit}/{exp}/stirring/target_rpm")
assert message is not None
assert float(message.payload) == 500
def test_pause_stirring_mid_cycle():
original_dc = 50
st = Stirrer(original_dc, unit, exp)
assert st.duty_cycle == original_dc
pause()
publish(f"pioreactor/{unit}/{exp}/stirring/$state/set", "sleeping")
pause()
assert st.duty_cycle == 0
publish(f"pioreactor/{unit}/{exp}/stirring/$state/set", "ready")
pause()
assert st.duty_cycle == 50
def test_publish_measured_rpm() -> None:
publish(f"pioreactor/{unit}/{exp}/stirring/measured_rpm",
None,
retain=True)
pause()
target_rpm = 500
with Stirrer(target_rpm, unit, exp,
rpm_calculator=RpmFromFrequency()) as st:
st.start_stirring()
assert st.target_rpm == target_rpm
pause()
message = subscribe(f"pioreactor/{unit}/{exp}/stirring/measured_rpm")
assert message is not None
assert json.loads(message.payload)["rpm"] == 0
def test_pause_stirring_mid_cycle() -> None:
with Stirrer(500, unit, exp, rpm_calculator=None) as st:
original_dc = st.duty_cycle
pause()
publish(f"pioreactor/{unit}/{exp}/stirring/$state/set", "sleeping")
pause()
pause()
pause()
assert st.state == st.SLEEPING
assert st.duty_cycle == 0
publish(f"pioreactor/{unit}/{exp}/stirring/$state/set", "ready")
pause()
pause()
pause()
assert st.state == st.READY
assert st.duty_cycle == original_dc
def test_change_target_rpm_mid_cycle() -> None:
original_rpm = 500
with Stirrer(original_rpm, unit, exp,
rpm_calculator=RpmCalculator()) as st:
assert st.target_rpm == original_rpm
pause()
new_rpm = 750
publish(f"pioreactor/{unit}/{exp}/stirring/target_rpm/set", new_rpm)
pause()
assert st.target_rpm == new_rpm
assert st.state == "ready"
publish(f"pioreactor/{unit}/{exp}/stirring/target_rpm/set", 0)
pause()
assert st.target_rpm == 0
def test_change_stirring_mid_cycle():
original_dc = 50
st = Stirrer(original_dc, unit, exp)
assert st.duty_cycle == original_dc
pause()
new_dc = 75
publish(f"pioreactor/{unit}/{exp}/stirring/duty_cycle/set", new_dc)
pause()
assert st.duty_cycle == new_dc
assert st.state == "ready"
publish(f"pioreactor/{unit}/{exp}/stirring/duty_cycle/set", 0)
pause()
assert st.duty_cycle == 0
pause()