Skip to content

Commit 57f6d94

Browse files
committed
Improved tests [skip ci]
1 parent 061de10 commit 57f6d94

1 file changed

Lines changed: 8 additions & 4 deletions

File tree

tests/test_pg8000.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)