def test_run_divide_zero(self):
args = MockArgs(divide=True, operand=[3, 0])
with self.assertRaises(ZeroDivisionError):
run(args)
def test_run_divide(self):
args = MockArgs(divide=True, operand=[8, 2])
self.assertEqual(run(args), "8 / 2 = 4.0")
def test_run_subtract(self):
args = MockArgs(subtract=True, operand=[6, 4])
self.assertEqual(run(args), "6 - 4 = 2")
def test_run_multiply(self):
args = MockArgs(multiply=True, operand=[2, 6])
self.assertEqual(run(args), "2 * 6 = 12")
def test_run_add(self):
args = MockArgs(add=True, operand=[3, 5])
self.assertEqual(run(args), "3 + 5 = 8")