def test_set_state(self):
relay_group = RelayGroup(2)
relay0 = Relay('relay0', relay_group, 0)
relay1 = Relay('relay1', relay_group, 1)
assert relay0.state == Relay.OPEN
assert relay1.state == Relay.OPEN
relay0.state = Relay.CLOSED
assert relay0.state == Relay.CLOSED
assert relay1.state == Relay.OPEN
relay1.state = Relay.CLOSED
assert relay0.state == Relay.CLOSED
assert relay1.state == Relay.CLOSED
def test_write_states(self):
hardware_device = Mock()
relay_group = RelayGroup(8, hardware_device)
relay0 = Relay('relay0', relay_group, 0)
relay1 = Relay('relay1', relay_group, 1)
relay2 = Relay('relay2', relay_group, 2)
relay0.close()
hardware_device.write.assert_called_with(1)
relay1.close()
hardware_device.write.assert_called_with(3)
relay2.close()
hardware_device.write.assert_called_with(7)
relay1.open()
hardware_device.write.assert_called_with(5)