Added test for AsIs

ankane · ankane · commit 01159f5f8133 · 2023-09-15T15:53:26.000-07:00
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]))
