Merge pull request #7 from hayabhay/tests

Added extremely important tests to make this puppy production ready
jdagdelen · Apr 18, 2023 · ce0d6b1 · ce0d6b1
2 parents f528e45 + 8569132
commit ce0d6b1
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1 +1,3 @@
 numpy
+pytest
+openai
diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/test_galaxy_brain_math_shit.py b/tests/test_galaxy_brain_math_shit.py
@@ -0,0 +1,14 @@
+import numpy as np
+from hyperdb import euclidean_metric
+
+# Test totally written by me and not gpt/co-pilot
+def test_euclid_bros_metric():
+    data_vectors = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
+    query_vector = np.array([1, 1, 1])
+    result = euclidean_metric(data_vectors, query_vector)
+
+    # All important test to see if shapes are based
+    try:
+        assert result.shape == (3,)
+    except AssertionError:
+        raise AssertionError("Oh my god, you killed Euclid! You bas-turd!")