Upload New File

tests/distance_matrix_test.py

+from knn import *
+import numpy as np
+
+def distance_matrix_test():
+    train = np.random.rand(100, 1000)
+    test = np.random.rand(100, 1000)
+    dists = distance_matrix(train, test)
+    assert dists.shape == (100, 80)
+
+distance_matrix_test()