diff --git a/test_utils.py b/test_utils.py index 1ec3877..18da312 100644 --- a/test_utils.py +++ b/test_utils.py @@ -1,4 +1,9 @@ -from utils import funny +from utils import funny, funny2 + def test_funny(): assert funny(2, 1) == 3 + + +def test_funny2(): + assert funny2(2, 2) == 5 diff --git a/utils.py b/utils.py index 696a495..a522d79 100644 --- a/utils.py +++ b/utils.py @@ -1,2 +1,6 @@ def funny(x, y): return x + y + + +def funny2(x, y): + return x**y