Update test_select.py

infiniflow · Sep 20, 2024 · 297d68c · 297d68c
1 parent ac1c408
commit 297d68c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/test_pysdk/test_select.py b/python/test_pysdk/test_select.py
@@ -29,9 +29,13 @@ def http(request):
 @pytest.fixture(scope="class")
 def setup_class(request, local_infinity, http):
     if local_infinity:
+        module = importlib.import_module("infinity_embedded.index")
+        globals()["index"] = module
         module = importlib.import_module("infinity_embedded.common")
         func = getattr(module, 'ConflictType')
         globals()['ConflictType'] = func
+        func = getattr(module, 'InfinityException')
+        globals()['InfinityException'] = func
         uri = common_values.TEST_LOCAL_PATH
         request.cls.infinity_obj = infinity_embedded.connect(uri)
     elif http: