Improved tests [skip ci]

Author: ankane

tests/test_psycopg.py

@@ -75,13 +75,15 @@ def test_halfvec_binary_format(self):
         embedding = HalfVector([1.5, 2, 3])
         res = conn.execute('SELECT %b::halfvec', (embedding,), binary=True).fetchone()[0]
         assert res == HalfVector([1.5, 2, 3])
+        # TODO move
         assert res.to_list() == [1.5, 2, 3]
         assert np.array_equal(res.to_numpy(), np.array([1.5, 2, 3]))
 
     def test_halfvec_text_format(self):
         embedding = HalfVector([1.5, 2, 3])
         res = conn.execute('SELECT %t::halfvec', (embedding,)).fetchone()[0]
         assert res == HalfVector([1.5, 2, 3])
+        # TODO move
         assert res.to_list() == [1.5, 2, 3]
         assert np.array_equal(res.to_numpy(), np.array([1.5, 2, 3]))
 
@@ -114,6 +116,7 @@ def test_sparsevec_binary_format(self):
         embedding = SparseVector([1.5, 0, 2, 0, 3, 0])
         res = conn.execute('SELECT %b::sparsevec', (embedding,), binary=True).fetchone()[0]
         assert res == embedding
+        # TODO move
         assert res.dimensions() == 6
         assert res.indices() == [0, 2, 4]
         assert res.values() == [1.5, 2, 3]
@@ -124,6 +127,7 @@ def test_sparsevec_text_format(self):
         embedding = SparseVector([1.5, 0, 2, 0, 3, 0])
         res = conn.execute('SELECT %t::sparsevec', (embedding,)).fetchone()[0]
         assert res == embedding
+        # TODO move
         assert res.dimensions() == 6
         assert res.indices() == [0, 2, 4]
         assert res.values() == [1.5, 2, 3]
@@ -166,8 +170,8 @@ def test_binary_copy_to(self):
         cur = conn.cursor()
         with cur.copy("COPY psycopg_items (embedding, half_embedding) TO STDOUT WITH (FORMAT BINARY)") as copy:
             for row in copy.rows():
-                assert Vector.from_binary(row[0]).to_list() == [1.5, 2, 3]
-                assert HalfVector.from_binary(row[1]).to_list() == [1.5, 2, 3]
+                assert np.array_equal(Vector.from_binary(row[0]).to_numpy(), embedding)
+                assert HalfVector.from_binary(row[1]) == half_embedding
 
     def test_binary_copy_to_set_types(self):
         embedding = np.array([1.5, 2, 3])
@@ -178,7 +182,7 @@ def test_binary_copy_to_set_types(self):
             copy.set_types(['vector', 'halfvec'])
             for row in copy.rows():
                 assert np.array_equal(row[0], embedding)
-                assert row[1].to_list() == [1.5, 2, 3]
+                assert row[1] == half_embedding
 
     def test_vector_array(self):
         embeddings = [np.array([1.5, 2, 3]), np.array([4.5, 5, 6])]