def test_compute(self):
i = [1, 0, 0, 0, 99]
self.assertEqual(computer.compute(i), [2, 0, 0, 0, 99])
def test_compute_3(self):
i = [2, 4, 4, 5, 99, 0]
self.assertEqual(computer.compute(i), [2, 4, 4, 5, 99, 9801])
def test_compute_4(self):
i = [1, 1, 1, 4, 99, 5, 6, 0, 99]
self.assertEqual(computer.compute(i), [30, 1, 1, 4, 2, 5, 6, 0, 99])
def test_compute_2(self):
i = [2, 3, 0, 3, 99]
self.assertEqual(computer.compute(i), [2, 3, 0, 6, 99])
