testing pytest with functions

Author: Kinetics20

example_5.py

@@ -1,14 +1,2 @@
 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

@@ -3,3 +3,14 @@
 
 def test_divide_positive_numbers():
     assert divide(10, 2) == 5.0
+
+
+def test_divide_numbers_negative():
+    assert divide(10, 2) != 15.0
+
+
+try:
+    divide(10, 0)
+    raise AssertionError('STH is wrong')
+except ZeroDivisionError:
+    print('OK')