Improved tests [skip ci]

Author: ankane

tests/test_psycopg2.py

@@ -82,17 +82,15 @@ def test_halfvec_array(self):
 
         cur.execute('SELECT half_embeddings FROM psycopg2_items ORDER BY id')
         res = cur.fetchone()
-        assert res[0][0] == HalfVector([1.5, 2, 3])
-        assert res[0][1] == HalfVector([4.5, 5, 6])
+        assert res[0] == [HalfVector([1.5, 2, 3]), HalfVector([4.5, 5, 6])]
 
     def test_sparsevec_array(self):
         embeddings = [SparseVector([1.5, 2, 3]), SparseVector([4.5, 5, 6])]
         cur.execute('INSERT INTO psycopg2_items (sparse_embeddings) VALUES (%s::sparsevec[])', (embeddings,))
 
         cur.execute('SELECT sparse_embeddings FROM psycopg2_items ORDER BY id')
         res = cur.fetchone()
-        assert res[0][0] == SparseVector([1.5, 2, 3])
-        assert res[0][1] == SparseVector([4.5, 5, 6])
+        assert res[0] == [SparseVector([1.5, 2, 3]), SparseVector([4.5, 5, 6])]
 
     def test_cursor_factory(self):
         for cursor_factory in [DictCursor, RealDictCursor, NamedTupleCursor]:

tests/test_sqlalchemy.py

@@ -551,8 +551,7 @@ def test_halfvec_array(self, engine):
 
             # this fails if the driver does not cast arrays
             item = session.get(Item, 1)
-            assert item.half_embeddings[0] == HalfVector([1, 2, 3])
-            assert item.half_embeddings[1] == HalfVector([4, 5, 6])
+            assert item.half_embeddings == [HalfVector([1, 2, 3]), HalfVector([4, 5, 6])]
 
 
 @pytest.mark.parametrize('engine', async_engines)