testing pytest with functions

Author: Kinetics20

example_4.py

@@ -7,7 +7,6 @@ def divide(a, b):
 else:
     raise AssertionError('STH is wrong')
 
-
 if divide(10, 2) != 15.0:
     print('OK')
 else:

example_5.py

@@ -0,0 +1,14 @@
+def divide(a, b):
+    return a / b
+
+
+if divide(10, 2) != 15.0:
+    print('OK')
+else:
+    raise AssertionError('STH is wrong')
+
+try:
+    divide(10, 0)
+    raise AssertionError('STH is wrong')
+except ZeroDivisionError:
+    print('OK')

test_example_5.py

@@ -0,0 +1,5 @@
+from example_5 import divide
+
+
+def test_divide_positive_numbers():
+    assert divide(10, 2) == 5.0