diff --git a/test_unnecessarymath.py b/test_unnecessarymath.py index c996ff315a554d02cdece145357a39e1fdef0f85..31220b22bacbd0ec376f0904f39cd0227a0511a3 100644 --- a/test_unnecessarymath.py +++ b/test_unnecessarymath.py @@ -8,3 +8,8 @@ def test_sqrt(): assert um.sqrt(1.) == 1. assert um.sqrt(0.) == 0. assert um.sqrt(4.) == 2. + +def power_test(): + assert um.power(3, 3) == 27 + assert um.power(2, 4) == 16 + assert um.power(5, 3) == 125 \ No newline at end of file diff --git a/unnecessarymath.py b/unnecessarymath.py index 71d4ad67992d7a7a98c261f5e7d053d2b4eea0b9..246be46e894752ec57590ce7396d55ac15286139 100644 --- a/unnecessarymath.py +++ b/unnecessarymath.py @@ -23,3 +23,14 @@ def square(x): 16.0 """ return x * x + +def power(x, y): + """ + Evaluates a number x to the power y + + >>> power(2, 3) + 8 + >>> power(3, 2) + 9 + """ + return x ** y \ No newline at end of file