add pytest

py/pytest_mini_project/simplemath.py

@@ -0,0 +1,15 @@
+# simplemath.py
+
+def add(a, b):
+    return a + b
+
+def subtract(a, b):
+    return a - b
+
+def multiply(a, b):
+    return a * b
+
+def divide(a, b):
+    if b == 0:
+        raise ValueError("Cannot divide by zero")
+    return a / b

py/pytest_mini_project/test_simplemath.py

@@ -0,0 +1,20 @@
+# test_simplemath.py
+
+import pytest
+from pytest_mini_project.simplemath import add, subtract, multiply, divide
+
+def test_add():
+    assert add(1, 2) == 3
+
+def test_subtract():
+    assert subtract(2, 1) == 1
+
+def test_multiply():
+    assert multiply(2, 3) == 6
+
+def test_divide():
+    assert divide(6, 2) == 3
+
+def test_divide_by_zero():
+    with pytest.raises(ValueError):
+        divide(1, 0)