def test_simple(self): p = Program("4, 0") p.step() self.assertEqual(p._Program__pos, 2) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [4, 0]) self.assertEqual(next(p), 4)
def test_day9_part1(self): datas = [ ( "109,1,204,-1,1001,100,1,100,1008,100,16,101,1006,101,0,99", [ 109, 1, 204, -1, 1001, 100, 1, 100, 1008, 100, 16, 101, 1006, 101, 0, 99, ], ), ("1102,34915192,34915192,7,4,7,99,0", [1219070632396864]), ("104,1125899906842624,99", [1125899906842624]), ] for data, result in datas: with self.subTest(data=data): p = Program(data) p.execute() self.assertEqual(list(p), result)
def test_direct(self): p = Program("104, 0") p.step() self.assertEqual(p._Program__pos, 2) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [104, 0]) self.assertEqual(next(p), 0)
def test_simple(self): p = Program("3, 1") p.send_input(42) p.step() self.assertEqual(p._Program__pos, 2) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [3, 42])
def test_day2_part1(self): datas = [ ("1,9,10,3,2,3,11,0,99,30,40,50", 3500), ] for data, result in datas: with self.subTest(data=data): p = Program(data) p.execute() self.assertEqual(p._Program__memory[0], result)
def test_day5_part2(self): datas = [ ( "3,21,1008,21,8,20,1005,20,22,107,8,21,20,1006,20,31,1106,0,36,98,0,0,1002,21,125,20,4,20,1105,1,46,104,999,1105,1,46,1101,1000,1,20,4,20,1105,1,46,98,99", 7, 999, ), ( "3,21,1008,21,8,20,1005,20,22,107,8,21,20,1006,20,31,1106,0,36,98,0,0,1002,21,125,20,4,20,1105,1,46,104,999,1105,1,46,1101,1000,1,20,4,20,1105,1,46,98,99", 8, 1000, ), ( "3,21,1008,21,8,20,1005,20,22,107,8,21,20,1006,20,31,1106,0,36,98,0,0,1002,21,125,20,4,20,1105,1,46,104,999,1105,1,46,1101,1000,1,20,4,20,1105,1,46,98,99", 9, 1001, ), ] for data, data_input, result in datas: with self.subTest(data=data): p = Program(data) p.send_input(data_input) p.execute() self.assertEqual(next(p), result)
def test_immediate_result_error(self): p = Program("10001, 0, 0, 0, 99") with self.assertRaises(AssertionError): p.step()
def test_immediate_two(self): p = Program("1102, 5, 4, 0, 99") p.step() self.assertEqual(p._Program__pos, 4) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [20, 5, 4, 0, 99])
def test_inplace(self): p = Program("2, 0, 0, 0, 99") p.step() self.assertEqual(p._Program__pos, 4) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [4, 0, 0, 0, 99])
def test_false(self): p = Program("8, 0, 1, 3") p.step() self.assertEqual(p._Program__pos, 4) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [8, 0, 1, 0])
def test_direct_true(self): p = Program("1107, 0, 1, 3") p.step() self.assertEqual(p._Program__pos, 4) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [1107, 0, 1, 1])
def test_true(self): p = Program("7, 1, 0, 3") p.step() self.assertEqual(p._Program__pos, 4) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [7, 1, 0, 1])
def test_simple_true(self): p = Program("6, 1, 0") p.step() self.assertEqual(p._Program__pos, 3) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [6, 1, 0])
def test_simple_false(self): p = Program("5, 2, 0") p.step() self.assertEqual(p._Program__pos, 3) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [5, 2, 0])
def test_direct_true(self): p = Program("1105, 1, 0") p.step() self.assertEqual(p._Program__pos, 0) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [1105, 1, 0])
def test_out_of_range(self): p = Program("98, 0, 0, 100, 99") with self.assertRaises(AssertionError): p.step()
def test_simple(self): p = Program("2, 4, 5, 6, 1, 1, 0") p.step() self.assertEqual(p._Program__pos, 4) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [2, 4, 5, 6, 1, 1, 1])
def test_direct_false(self): p = Program("1108, 1, 0, 3") p.step() self.assertEqual(p._Program__pos, 4) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [1108, 1, 0, 0])
def test_immediate_one(self): p = Program("1002, 0, 4, 0, 99") p.step() self.assertEqual(p._Program__pos, 4) self.assertEqual(p._Program__running, True) self.assertEqual(p._Program__memory, [4008, 0, 4, 0, 99])
def test_simple(self): p = Program("99") p.step() self.assertEqual(p._Program__pos, 1) self.assertEqual(p._Program__running, False) self.assertEqual(p._Program__memory, [99])
def main(): data = sys.stdin.readline() p = Program(data) p.send_input(1) p.execute() print(list(p))
def test_out_of_range(self): p = Program("1, 0, 0, 100, 99") with self.assertRaises(IndexError): p.step()