1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
#!/usr/bin/python import unittest import main class ExamplesPart1(unittest.TestCase): def test_example1(self): key = "abcdef" self.assertEqual(main.solve_partn(1, key), 609043) def test_example2(self): key = "pqrstuv" self.assertEqual(main.solve_partn(1, key), 1048970) if __name__ == "__main__": unittest.main()