diff --git a/src/main/py/calculations_with_roots.py b/src/main/py/calculations_with_roots.py index 3cf4ee6..fb73efe 100644 --- a/src/main/py/calculations_with_roots.py +++ b/src/main/py/calculations_with_roots.py @@ -1,2 +1,4 @@ +import math + def multiplyRoots(first_number, second_number): - return 0 \ No newline at end of file + return math.sqrt(first_number * second_number) \ No newline at end of file diff --git a/src/test/py/test_calculations_with_roots.py b/src/test/py/test_calculations_with_roots.py index 713dcc2..17fb5c2 100644 --- a/src/test/py/test_calculations_with_roots.py +++ b/src/test/py/test_calculations_with_roots.py @@ -12,6 +12,9 @@ class calculationsWithRoots(unittest.TestCase): def test_sqrt_0_times_sqrt_8_should_be_0(self): self.assertEqual(multiplyRoots(0, 8), 0) + def test_sqrt_2_times_sqrt_8_should_be_4(self): + self.assertEqual(multiplyRoots(2, 8), 4) + if __name__ == '__main__': unittest.main()