class KeypadTest(unittest.TestCase): def setUp(self): self.keypad = Keypad() # ULL # RRDDD # LURDL # UUUUD # # So, in this example, the bathroom code is 1985. def test_sample(self): self.assertEqual(self.keypad.executeRow('ULL'), '1') self.assertEqual(self.keypad.executeRow('RRDDD'), '9') self.assertEqual(self.keypad.executeRow('LURDL'), '8') self.assertEqual(self.keypad.executeRow('UUUUD'), '5')
from keypad import Keypad with open('../input') as handle: keypad = Keypad() code = [] for row in handle: code.append(keypad.executeRow(row)) print "Door Code:", ''.join(code)