test

Practical_MLOps/Ch1/hello.py

@@ -1,7 +1,15 @@
+import numpy as np 
+
 def add(x, y):
     """This is an add function"""
     return x + y
 
+def numpy_add(x, y):    
+    return x + y 
+
+
 print(add(1, 1))
 
-print(add(2, 1))
+print(add(2, 1))
+
+print(numpy_add(np.arange(4), np.arange(4)))

Practical_MLOps/Ch1/test_hello.py

@@ -1,5 +1,9 @@
-from hello import add 
+import numpy as np 
+from hello import add, numpy_add
+
 
 def test_add():
     assert 2 == add(1, 1)
 
+def test_numpy_add():
+    assert 1 == 1

requirements.txt

@@ -4,3 +4,4 @@ pandas
 pytest 
 pytest-cov
 pylint
+numpy