functions and tests:

Author: davidslater

requirements.txt

@@ -0,0 +1 @@
+numpy

some_functions.py

@@ -0,0 +1,13 @@
+"""
+This module uses numpy but is otherwise very easy to understand and test.
+"""
+
+import numpy as np
+
+
+def dot_multiply(a, b):
+    return np.dot(a, b)
+
+
+def two_norm(array):
+    return np.linalg.norm(array, ord=2)

test_some_functions.py

@@ -0,0 +1,12 @@
+from some_functions import *
+
+
+def test_dot_multiply():
+    a = np.array([1, 2, 3])
+    b = np.array([4, 5, 6])
+    assert dot_multiply(a, b) == 32
+
+
+def test_two_norm():
+    array = np.array([3, 4])
+    assert two_norm(array) == 5