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