def test_force():
"""
lib/armedswitch.py: test force()
"""
s = ArmedCheckSwitch()
assert s.is_switched() == False
assert s.switch() == True
assert s.is_armed() == False
assert s.switch() == False
assert s.is_switched() == True
s.force(False)
assert s.is_switched() == False
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
