def test_wait(self): code = [3, 3, 99, 0] amplifier = Amplifier(code) status, signal = amplifier.execute() self.assertEqual(status, 'WAITING') self.assertEqual(signal, None) status, signal = amplifier.execute(10) self.assertEqual(status, 'HALT') self.assertEqual(signal, None)
def test_sample_9(self): code = [3, 3, 1107, -1, 8, 3, 4, 3, 99] amplifier = Amplifier(code) status, signal = amplifier.execute(0) self.assertEqual(status, 'OUTPUT') self.assertEqual(signal, 1) status, signal = amplifier.execute() self.assertEqual(status, 'HALT') self.assertEqual(signal, None)
def test_sample_11(self): code = [3, 12, 6, 12, 15, 1, 13, 14, 13, 4, 13, 99, -1, 0, 1, 9] amplifier = Amplifier(code) status, signal = amplifier.execute(1) self.assertEqual(status, 'OUTPUT') self.assertEqual(signal, 1) status, signal = amplifier.execute() self.assertEqual(status, 'HALT') self.assertEqual(signal, None)
def test_io(self): code = [3, 9, 1002, 9, 2, 10, 4, 10, 99, 0, 0] amplifier = Amplifier(code) status, signal = amplifier.execute(3) self.assertEqual(status, 'OUTPUT') self.assertEqual(signal, 6) status, signal = amplifier.execute() self.assertEqual(status, 'HALT') self.assertEqual(signal, None)