You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18 lines
669 B
18 lines
669 B
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() |