class CalculatorTest(unittest.TestCase): def setUp(self): self.calc = Calculator() def test_add(self): result = self.calc.add(5, 3) self.assertEqual(result, 8) def test_subtract(self): result = self.calc.subtract(10, 4) self.assertEqual(result, 6) def multiply_test(self): result = self.calc.multiply(9, 2) self.assertEqual(result, 18) def test_divide(self): result = self.calc.divide(6, 3) self.assertEqual(result, 2) with self.assertRaises(ZeroDivisionError): self.calc.divide(10, 0) self.calc.divide(0, 0)