diff --git a/app/calculator.py b/app/calculator.py index 668689d3..e769ea61 100644 --- a/app/calculator.py +++ b/app/calculator.py @@ -16,3 +16,6 @@ def divide(x, y): def abs(x): return x if x >= 0 else -x + + def pow(x, y): + return x**y diff --git a/app/test_calculator.py b/app/test_calculator.py index 27fa8ef6..3ca4cedc 100644 --- a/app/test_calculator.py +++ b/app/test_calculator.py @@ -38,3 +38,9 @@ def test_abs(): assert Calculator.abs(1) == 1 assert Calculator.abs(-1) == 1 assert Calculator.abs(0) == 0 + + +def test_pow(): + assert Calculator.pow(2, 3) == 8 + assert Calculator.pow(2, 0) == 1 + assert Calculator.pow(2, -1) == 0.5