Create a division function

gorqflow/playtools/__init__.py

@@ -1,2 +1,2 @@
 from ._playtools import update_dict, print_dict
-from ._math import add_one, sub_one
+from ._math import add_one, sub_one, divide

gorqflow/playtools/_math.py

@@ -20,3 +20,9 @@ def sub_one(x: int) -> int:
         int: x - 1
     """
     return x - 1
+
+
+def divide(nominator: int | float, denominator: int | float) -> float:
+    if denominator == 0:
+        raise ZeroDivisionError("Cannot divide by zero")
+    return nominator / denominator

tests/playtools/test_math.py

@@ -1,4 +1,5 @@
-from gorqflow.playtools import add_one, sub_one
+import pytest
+from gorqflow.playtools import add_one, sub_one, divide
 
 
 def test_add_one():
@@ -7,3 +8,12 @@ def test_add_one():
 
 def test_sub_one():
     assert sub_one(1) == 0
+
+
+def test_divide_ok():
+    assert divide(15., 3.) == 5.
+
+
+def test_divide_by_zero():
+    with pytest.raises(ZeroDivisionError):
+        divide(15., 0.)