Improved imports for tests [skip ci]

Author: ankane

tests/test_asyncpg.py

@@ -1,6 +1,7 @@
 import asyncpg
 import numpy as np
-from pgvector.asyncpg import register_vector, SparseVector
+from pgvector import SparseVector
+from pgvector.asyncpg import register_vector
 import pytest
 
 

tests/test_bit.py

@@ -1,5 +1,5 @@
 import numpy as np
-from pgvector.utils import Bit
+from pgvector import Bit
 import pytest
 
 

tests/test_django.py

@@ -12,7 +12,8 @@
 import numpy as np
 import os
 import pgvector.django
-from pgvector.django import VectorExtension, VectorField, HalfVectorField, BitField, SparseVectorField, IvfflatIndex, HnswIndex, L2Distance, MaxInnerProduct, CosineDistance, L1Distance, HammingDistance, JaccardDistance, HalfVector, SparseVector
+from pgvector import HalfVector, SparseVector
+from pgvector.django import VectorExtension, VectorField, HalfVectorField, BitField, SparseVectorField, IvfflatIndex, HnswIndex, L2Distance, MaxInnerProduct, CosineDistance, L1Distance, HammingDistance, JaccardDistance
 from unittest import mock
 
 settings.configure(

tests/test_half_vector.py

@@ -1,5 +1,5 @@
 import numpy as np
-from pgvector.utils import HalfVector
+from pgvector import HalfVector
 import pytest
 
 

tests/test_peewee.py

@@ -1,7 +1,8 @@
 from math import sqrt
 import numpy as np
 from peewee import Model, PostgresqlDatabase, fn
-from pgvector.peewee import VectorField, HalfVectorField, FixedBitField, SparseVectorField, SparseVector
+from pgvector import SparseVector
+from pgvector.peewee import VectorField, HalfVectorField, FixedBitField, SparseVectorField
 
 db = PostgresqlDatabase('pgvector_python_test')
 

tests/test_psycopg.py

@@ -1,5 +1,6 @@
 import numpy as np
-from pgvector.psycopg import register_vector, register_vector_async, Bit, HalfVector, SparseVector, Vector
+from pgvector import Bit, HalfVector, SparseVector, Vector
+from pgvector.psycopg import register_vector, register_vector_async
 import psycopg
 from psycopg_pool import ConnectionPool, AsyncConnectionPool
 import pytest

tests/test_psycopg2.py

@@ -1,5 +1,6 @@
 import numpy as np
-from pgvector.psycopg2 import register_vector, Vector, HalfVector, SparseVector
+from pgvector import HalfVector, SparseVector, Vector
+from pgvector.psycopg2 import register_vector
 import psycopg2
 from psycopg2.extras import DictCursor, RealDictCursor, NamedTupleCursor
 from psycopg2.pool import ThreadedConnectionPool

tests/test_sparse_vector.py

@@ -1,5 +1,5 @@
 import numpy as np
-from pgvector.utils import SparseVector
+from pgvector import SparseVector
 import pytest
 from scipy.sparse import coo_array
 

tests/test_sqlalchemy.py

@@ -1,7 +1,8 @@
 import asyncpg
 import numpy as np
 import os
-from pgvector.sqlalchemy import VECTOR, HALFVEC, BIT, SPARSEVEC, SparseVector, avg, sum
+from pgvector import SparseVector
+from pgvector.sqlalchemy import VECTOR, HALFVEC, BIT, SPARSEVEC, avg, sum
 import pytest
 from sqlalchemy import create_engine, event, insert, inspect, select, text, MetaData, Table, Column, Index, Integer, ARRAY
 from sqlalchemy.exc import StatementError

tests/test_sqlmodel.py

@@ -1,5 +1,6 @@
 import numpy as np
-from pgvector.sqlalchemy import VECTOR, HALFVEC, BIT, SPARSEVEC, SparseVector, avg, sum
+from pgvector import SparseVector
+from pgvector.sqlalchemy import VECTOR, HALFVEC, BIT, SPARSEVEC, avg, sum
 import pytest
 from sqlalchemy.exc import StatementError
 from sqlmodel import Field, Index, Session, SQLModel, create_engine, delete, select, text