fix redis_url usage in tests

Author: tylerhutcherson

tests/integration/test_connection.py

@@ -20,10 +20,6 @@
 EXPECTED_LIB_NAME = f"redis-py(redisvl_v{__version__})"
 
 
-def test_get_address_from_env(redis_url):
-    assert get_address_from_env() == redis_url
-
-
 def test_unpack_redis_modules():
     module_list = [
         {

tests/integration/test_llmcache.py

@@ -894,33 +894,35 @@ def test_bad_dtype_connecting_to_existing_cache(redis_url):
         )
 
 
-def test_vectorizer_dtype_mismatch():
+def test_vectorizer_dtype_mismatch(redis_url):
     with pytest.raises(ValueError):
         SemanticCache(
             name="test_dtype_mismatch",
             dtype="float32",
             vectorizer=HFTextVectorizer(dtype="float16"),
+            redis_url=redis_url,
             overwrite=True,
         )
 
 
-def test_invalid_vectorizer():
+def test_invalid_vectorizer(redis_url):
     with pytest.raises(TypeError):
         SemanticCache(
             name="test_invalid_vectorizer",
             vectorizer="invalid_vectorizer",  # type: ignore
+            redis_url=redis_url,
             overwrite=True,
         )
 
 
-def test_passes_through_dtype_to_default_vectorizer():
+def test_passes_through_dtype_to_default_vectorizer(redis_url):
     # The default is float32, so we should see float64 if we pass it in.
     cache = SemanticCache(
-        name="test_pass_through_dtype", dtype="float64", overwrite=True
+        name="test_pass_through_dtype", dtype="float64", redis_url=redis_url, overwrite=True
     )
     assert cache._vectorizer.dtype == "float64"
 
 
-def test_deprecated_dtype_argument():
+def test_deprecated_dtype_argument(redis_url):
     with pytest.warns(DeprecationWarning):
-        SemanticCache(name="test_deprecated_dtype", dtype="float32", overwrite=True)
+        SemanticCache(name="test_deprecated_dtype", dtype="float32", redis_url=redis_url, overwrite=True)

tests/integration/test_search_results.py

@@ -16,7 +16,7 @@ def filter_query():
 
 
 @pytest.fixture
-def index(sample_data):
+def index(sample_data, redis_url):
     fields_spec = [
         {"name": "credit_score", "type": "tag"},
         {"name": "user", "type": "tag"},
@@ -47,7 +47,7 @@ def index(sample_data):
     index = SearchIndex.from_dict(json_schema)
 
     # connect to local redis instance
-    index.connect(os.environ["REDIS_URL"])
+    index.connect(redis_url=redis_url)
 
     # create the index (no data yet)
     index.create(overwrite=True)

tests/integration/test_semantic_router.py

@@ -315,42 +315,46 @@ def test_bad_dtype_connecting_to_exiting_router(redis_url, routes):
         )
 
 
-def test_vectorizer_dtype_mismatch(routes):
+def test_vectorizer_dtype_mismatch(routes, redis_url):
     with pytest.raises(ValueError):
         SemanticRouter(
             name="test_dtype_mismatch",
             routes=routes,
             dtype="float32",
             vectorizer=HFTextVectorizer(dtype="float16"),
+            redis_url=redis_url,
             overwrite=True,
         )
 
 
-def test_invalid_vectorizer(routes):
+def test_invalid_vectorizer(routes, redis_url):
     with pytest.raises(TypeError):
         SemanticRouter(
             name="test_invalid_vectorizer",
             vectorizer="invalid_vectorizer",  # type: ignore
+            redis_url=redis_url,
             overwrite=True,
         )
 
 
-def test_passes_through_dtype_to_default_vectorizer(routes):
+def test_passes_through_dtype_to_default_vectorizer(routes, redis_url):
     # The default is float32, so we should see float64 if we pass it in.
     router = SemanticRouter(
         name="test_pass_through_dtype",
         routes=routes,
         dtype="float64",
+        redis_url=redis_url,
         overwrite=True,
     )
     assert router.vectorizer.dtype == "float64"
 
 
-def test_deprecated_dtype_argument(routes):
+def test_deprecated_dtype_argument(routes, redis_url):
     with pytest.warns(DeprecationWarning):
         SemanticRouter(
             name="test_deprecated_dtype",
             routes=routes,
             dtype="float32",
+            redis_url=redis_url,
             overwrite=True,
         )

tests/integration/test_session_manager.py

@@ -589,33 +589,35 @@ def test_bad_dtype_connecting_to_exiting_session(redis_url):
         )
 
 
-def test_vectorizer_dtype_mismatch():
+def test_vectorizer_dtype_mismatch(redis_url):
     with pytest.raises(ValueError):
         SemanticSessionManager(
             name="test_dtype_mismatch",
             dtype="float32",
             vectorizer=HFTextVectorizer(dtype="float16"),
+            redis_url=redis_url,
             overwrite=True,
         )
 
 
-def test_invalid_vectorizer():
+def test_invalid_vectorizer(redis_url):
     with pytest.raises(TypeError):
         SemanticSessionManager(
             name="test_invalid_vectorizer",
             vectorizer="invalid_vectorizer",  # type: ignore
+            redis_url=redis_url,
             overwrite=True,
         )
 
 
-def test_passes_through_dtype_to_default_vectorizer():
+def test_passes_through_dtype_to_default_vectorizer(redis_url):
     # The default is float32, so we should see float64 if we pass it in.
     cache = SemanticSessionManager(
-        name="test_pass_through_dtype", dtype="float64", overwrite=True
+        name="test_pass_through_dtype", dtype="float64", redis_url=redis_url, overwrite=True
     )
     assert cache._vectorizer.dtype == "float64"
 
 
-def test_deprecated_dtype_argument():
+def test_deprecated_dtype_argument(redis_url):
     with pytest.warns(DeprecationWarning):
-        SemanticSessionManager(name="float64 session", dtype="float64", overwrite=True)
+        SemanticSessionManager(name="float64 session", dtype="float64", redis_url=redis_url, overwrite=True)