diff --git a/requirements.txt b/requirements.txt index bffa19b..c52d322 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1,3 @@ numpy +pytest +openai \ No newline at end of file diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/test_galaxy_brain_math_shit.py b/tests/test_galaxy_brain_math_shit.py new file mode 100644 index 0000000..0547eb1 --- /dev/null +++ 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!") \ No newline at end of file