diff --git a/calculator/tests/unit_tests_calculator.py b/calculator/tests/unit_tests_calculator.py index 3573845..e9d6e09 100644 --- a/calculator/tests/unit_tests_calculator.py +++ b/calculator/tests/unit_tests_calculator.py @@ -1,6 +1,6 @@ # test_calculator.py -from calculator import add, multiply, divide, subtract, square_root +from calculator import add, multiply, divide, subtract, square_root, exponentiate def test_addition(): @@ -31,3 +31,9 @@ def test_square_root(): assert square_root(4) == 2 assert square_root(25) == 5 assert square_root(9) == 3 + + +def test_exponentiate(): + assert exponentiate(2, 2) == 4 + assert exponentiate(-3, 3) == -27 + assert exponentiate(9, 0) == 1