@@ -25,7 +25,8 @@ def setup_method(self):
2525
2626 def test_vector (self ):
2727 embedding = Vector ([1.5 , 2 , 3 ])
28- conn .run ('INSERT INTO pg8000_items (embedding) VALUES (:embedding), (NULL)' , embedding = embedding )
28+ embedding2 = None
29+ conn .run ('INSERT INTO pg8000_items (embedding) VALUES (:embedding), (:embedding2)' , embedding = embedding , embedding2 = embedding2 )
2930
3031 res = conn .run ('SELECT embedding FROM pg8000_items ORDER BY id' )
3132 assert res [0 ][0 ] == embedding
@@ -42,23 +43,26 @@ def test_vector_numpy(self):
4243
4344 def test_halfvec (self ):
4445 embedding = HalfVector ([1.5 , 2 , 3 ])
45- conn .run ('INSERT INTO pg8000_items (half_embedding) VALUES (:embedding), (NULL)' , embedding = embedding )
46+ embedding2 = None
47+ conn .run ('INSERT INTO pg8000_items (half_embedding) VALUES (:embedding), (:embedding2)' , embedding = embedding , embedding2 = embedding2 )
4648
4749 res = conn .run ('SELECT half_embedding FROM pg8000_items ORDER BY id' )
4850 assert res [0 ][0 ] == embedding
4951 assert res [1 ][0 ] is None
5052
5153 def test_bit (self ):
5254 embedding = '101'
53- conn .run ('INSERT INTO pg8000_items (binary_embedding) VALUES (:embedding), (NULL)' , embedding = embedding )
55+ embedding2 = None
56+ conn .run ('INSERT INTO pg8000_items (binary_embedding) VALUES (:embedding), (:embedding2)' , embedding = embedding , embedding2 = embedding2 )
5457
5558 res = conn .run ('SELECT binary_embedding FROM pg8000_items ORDER BY id' )
5659 assert res [0 ][0 ] == '101'
5760 assert res [1 ][0 ] is None
5861
5962 def test_sparsevec (self ):
6063 embedding = SparseVector ([1.5 , 2 , 3 ])
61- conn .run ('INSERT INTO pg8000_items (sparse_embedding) VALUES (:embedding), (NULL)' , embedding = embedding )
64+ embedding2 = None
65+ conn .run ('INSERT INTO pg8000_items (sparse_embedding) VALUES (:embedding), (:embedding2)' , embedding = embedding , embedding2 = embedding2 )
6266
6367 res = conn .run ('SELECT sparse_embedding FROM pg8000_items ORDER BY id' )
6468 assert res [0 ][0 ] == embedding
0 commit comments