def test_equals_2(self): self.assertEqual(lightswitch(4), 2)
def test_equals(self): self.assertEqual(lightswitch(3), 1)