Added tests for non-contiguous arrays [skip ci]

Author: ankane

examples/loading/example.py

@@ -1,4 +1,5 @@
 import numpy as np
+from pgvector import Vector
 from pgvector.psycopg import register_vector
 import psycopg
 
@@ -25,7 +26,7 @@
     copy.set_types(['vector'])
 
     for i, embedding in enumerate(embeddings):
-        copy.write_row([embedding])
+        copy.write_row([Vector(embedding)])
 
         # show progress
         if i % 10000 == 0:

examples/loading/pyproject.toml

@@ -9,3 +9,6 @@ dev = [
     "pgvector",
     "psycopg[binary]"
 ]
+
+[tool.uv.sources]
+pgvector = { path = "../..", editable = true }

tests/test_half_vector.py

@@ -34,6 +34,9 @@ def test_ndarray(self) -> None:
         assert HalfVector(arr).to_list() == [1, 2, 3]
         assert HalfVector(arr).to_numpy() is not arr
         assert HalfVector(arr).to_numpy().dtype == np.float16
+        # non-contiguous
+        assert HalfVector(np.flip(arr)).to_list() == [3, 2, 1]
+        assert HalfVector(np.flip(arr)).to_binary() == HalfVector([3, 2, 1]).to_binary()
 
     def test_int(self) -> None:
         with pytest.raises(ValueError) as error:

tests/test_vector.py

@@ -34,6 +34,9 @@ def test_ndarray(self) -> None:
         assert Vector(arr).to_list() == [1, 2, 3]
         assert Vector(arr).to_numpy() is not arr
         assert Vector(arr).to_numpy().dtype == np.float32
+        # non-contiguous
+        assert Vector(np.flip(arr)).to_list() == [3, 2, 1]
+        assert Vector(np.flip(arr)).to_binary() == Vector([3, 2, 1]).to_binary()
 
     def test_int(self) -> None:
         with pytest.raises(ValueError) as error: