import unittest from calculator import BaseCalculator, AdvancedCalculator class TestCalculator(unittest.TestCase): def test_base_calc(self): self.assertEqual(BaseCalculator.add(2, 3), 5) self.assertEqual(BaseCalculator.divide(10, 2), 5) with self.assertRaises(ZeroDivisionError): BaseCalculator.divide(10, 0) def test_advanced_calc(self): self.assertEqual(AdvancedCalculator.power(2, 3), 8) self.assertEqual(AdvancedCalculator.sqrt(16), 4) self.assertEqual(AdvancedCalculator.factorial(5), 120) self.assertTrue(AdvancedCalculator.is_prime(13)) if __name__ == "__main__": unittest.main()