fix(python): install the _zvec extension inside the zvec package (#511)

python/tests/test_convert.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import math
 
 import pytest
-from _zvec import _Doc
+from zvec._zvec import _Doc
 from zvec.model.convert import convert_to_py_doc, convert_to_cpp_doc
 from zvec import Doc, CollectionSchema, DataType, FieldSchema, VectorSchema
 

python/tests/test_doc.py

@@ -17,7 +17,7 @@ import math
 import pytest
 
 
-from _zvec import _Doc
+from zvec._zvec import _Doc
 from zvec import FieldSchema, VectorSchema, Doc, DataType
 
 

python/tests/test_fts_query.py

@@ -79,7 +79,7 @@ class TestFtsQueryBinding:
 
     def test_import_fts_query(self):
         """_Fts should be importable from _zvec.param."""
-        from _zvec.param import _Fts
+        from zvec._zvec.param import _Fts
 
         fts = _Fts()
         assert fts.query_string == ""
@@ -87,7 +87,7 @@ class TestFtsQueryBinding:
 
     def test_fts_query_set_fields(self):
         """Setting fields on _Fts should work."""
-        from _zvec.param import _Fts
+        from zvec._zvec.param import _Fts
 
         fts = _Fts()
         fts.query_string = "+hello -world"
@@ -99,7 +99,7 @@ class TestFtsQueryBinding:
 
     def test_fts_query_pickle(self):
         """_Fts should support pickling."""
-        from _zvec.param import _Fts
+        from zvec._zvec.param import _Fts
 
         fts = _Fts()
         fts.query_string = "+vector search"
@@ -112,7 +112,7 @@ class TestFtsQueryBinding:
 
     def test_search_query_fts_field(self):
         """_SearchQuery should have fts field."""
-        from _zvec.param import _Fts, _SearchQuery
+        from zvec._zvec.param import _Fts, _SearchQuery
 
         vq = _SearchQuery()
         # fts should be None by default (optional)
@@ -127,7 +127,7 @@ class TestFtsQueryBinding:
 
     def test_search_query_pickle_with_fts(self):
         """_SearchQuery with fts should survive pickling."""
-        from _zvec.param import _Fts, _SearchQuery
+        from zvec._zvec.param import _Fts, _SearchQuery
 
         vq = _SearchQuery()
         vq.topk = 10
@@ -145,7 +145,7 @@ class TestFtsQueryBinding:
 
     def test_search_query_pickle_without_fts(self):
         """_SearchQuery without fts should survive pickling."""
-        from _zvec.param import _SearchQuery
+        from zvec._zvec.param import _SearchQuery
 
         vq = _SearchQuery()
         vq.topk = 5

python/tests/test_params.py

@@ -40,7 +40,7 @@ from zvec import (
     VectorSchema,
 )
 
-from _zvec.param import _SearchQuery
+from zvec._zvec.param import _SearchQuery
 
 # ----------------------------
 # Invert Index Param Test Case

python/tests/test_query_executor.py

@@ -18,7 +18,7 @@ from unittest.mock import MagicMock, patch
 
 import numpy as np
 import math
-from _zvec.param import _SearchQuery
+from zvec._zvec.param import _SearchQuery
 
 import pytest
 from zvec.executor.query_executor import (