def test_constant_energy_no_change(transfocator): # tests constant_energy when there is no change to the req_energy pv initial_energy = 9536.5 transfocator.req_energy.put(initial_energy) transfocator.beam_energy.put(initial_energy) def nothing(transfocator): pass wrapped_func = constant_energy(nothing) wrapped_func(transfocator, 'req_energy', 0.1) wrapped_func(transfocator, 'beam_energy', 0.1)
def test_constant_energy_with_change(transfocator): # tests constant_energy when there is a signficant change to the req_energy # or beam_energy pv initial_energy = 9536.5 transfocator.req_energy.put(initial_energy) transfocator.beam_energy.put(initial_energy) new_energy = 9530.1 def change_energy(transfocator, new_energy): transfocator.req_energy.put(new_energy) transfocator.beam_energy.put(new_energy) with pytest.raises(TransfocatorEnergyInterrupt): wrapped_func = constant_energy(change_energy) wrapped_func(transfocator, 'req_energy', 0.1, new_energy) wrapped_func(transfocator, 'beam_energy', 0.1, new_energy)
def test_constant_energy_bad_input(transfocator): def nothing(transfocator): pass with pytest.raises(AttributeError): wrapped_func = constant_energy(nothing) wrapped_func(transfocator, 'bad_input_string', 0.1)