diff --git a/api/calculator/test_calculator.py b/api/calculator/test_calculator.py index ad1a589..225c700 100644 --- a/api/calculator/test_calculator.py +++ b/api/calculator/test_calculator.py @@ -28,4 +28,6 @@ def test_divide(): assert Calculator.divide(1.0, 2.0) == 0.5 assert Calculator.divide(0, 2.0) == 0 assert Calculator.divide(-4, 2.0) == -2.0 - \ No newline at end of file + +def test_divide_by_zero(): + assert Calculator.divide(1, 0) == "Cannot divide by 0" \ No newline at end of file