Added test for AsIs

pgvector · Sep 15, 2023 · 01159f5 · 01159f5
1 parent 695557e
commit 01159f5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_peewee.py b/tests/test_peewee.py
@@ -1,6 +1,6 @@
 from math import sqrt
 import numpy as np
-from peewee import Model, PostgresqlDatabase
+from peewee import AsIs, Model, PostgresqlDatabase
 from pgvector.peewee import VectorField
 
 db = PostgresqlDatabase('pgvector_python_test')
@@ -72,3 +72,4 @@ def test_where(self):
     def test_get_or_create(self):
         Item.get_or_create(id=1, defaults={'embedding': [1, 2, 3]})
         Item.get_or_create(embedding=np.array([4, 5, 6]))
+        Item.get_or_create(embedding=AsIs([7, 8, 9]))