def test_arm():
"""
armedcheckswitch.py: Test arm()
"""
s = ArmedCheckSwitch()
assert s.is_armed() == True
assert s.switch() == True
assert s.is_armed() == False
s.arm()
assert s.is_armed() == True
def test_switch():
"""
armedcheckswitch.py: Test switch()
"""
s = ArmedCheckSwitch(on_check=is_equal_to_one, off_check=is_equal_to_two)
assert s.is_switched() == False
assert s.switch(4) == False
assert s.is_switched() == False
assert s.switch(1) == True
assert s.is_switched() == True
# Should need to re-arm before switching again
assert s.switch(2) == False
assert s.is_switched() == True
s.arm()
assert s.switch(2) == True
assert s.is_switched() == False
def test_switch_on_off():
"""
armedcheckswitch.py: Test switch_on() and switch_off()
"""
s = ArmedCheckSwitch(on_check=is_equal_to_one, off_check=is_equal_to_two)
assert s.is_switched() == False
assert s.switch_on(3) == False
assert s.is_switched() == False
assert s.switch_on(1) == True
assert s.is_switched() == True
assert s.switch_off(1) == False
assert s.is_switched() == True
# Should need to re-arm the switch to turn it off
assert s.switch_off(2) == False
assert s.is_switched() == True
s.arm()
assert s.switch_off(2) == True
assert s.is_switched() == False
