Eobrien2502/2 zero division exception test

pythonmaths/arithmetic.py

@@ -51,8 +51,15 @@ def divide(x: int | float, y: int | float) -> float:
         5.0
     >>> arithmetic.divide(5, 2)
         2.5
+
+    I've added a try and except code block to catch a divide by zero error.
     """
-    return x / y
+    try:
+        return x / y
+    except ZeroDivisionError as e:
+        raise ZeroDivisionError(
+            "You can not divide by 0, please choose another value for 'y'."
+        ) from e
 
 
 def multiply(x: int | float, y: int | float) -> float:

tests/test_arithmetic.py

@@ -64,3 +64,10 @@ def test_multiply(x: int | float, y: int | float, expected: int | float) -> None
 def test_divide(x: int | float, y: int | float, expected: int | float) -> None:
     """Test the divide function."""
     assert arithmetic.divide(x, y) == pytest.approx(expected)
+
+def test_divide_zero_division_exception() -> None:
+    """Test that a ZeroDivisionError is raised by the divide() function."""
+    with pytest.raises(ZeroDivisionError):
+        arithmetic.divide(2, 0)
+
+