cleanup testcontainer implementation

Author: tylerhutcherson

conftest.py

@@ -1,20 +1,17 @@
 import os
 import pytest
-import asyncio
 
 from redisvl.redis.connection import RedisConnectionFactory
 from testcontainers.compose import DockerCompose
 
 
-
 @pytest.fixture(scope="session", autouse=True)
 def redis_container(request):
     """
     Create a unique Compose project for each xdist worker by setting
     COMPOSE_PROJECT_NAME. That prevents collisions on container/volume names.
     """
-    # In xdist, the config has "workerid" in workerinput.  For the main (non-xdist)
-    # process, 'workerid' is often 'master' or something similar.
+    # In xdist, the config has "workerid" in workerinput
     worker_id = request.config.workerinput.get("workerid", "master")
 
     # Set the Compose project name so containers do not clash across workers
@@ -26,21 +23,11 @@ def redis_container(request):
         compose_file_name="docker-compose.yml",
         pull=True,
     )
-
     compose.start()
 
-    # If you mapped the container port 6379:6379 in docker-compose.yml,
-    # you might have collisions across workers. If you rely on ephemeral
-    # host ports, remove the `ports:` block in docker-compose.yml and do:
-    redis_host, redis_port = compose.get_service_host_and_port("redis", 6379)
-    #redis_url = f"redis://{redis_host}:{redis_port}"
-    #os.environ["REDIS_URL"] = redis_url
-
     yield compose
 
     compose.stop()
-    # Optionally, clean up the COMPOSE_PROJECT_NAME you set:
-    os.environ.pop("COMPOSE_PROJECT_NAME", None)
 
 
 @pytest.fixture(scope="session")

scripts.py

@@ -29,7 +29,7 @@ def check_mypy():
 
 
 def test():
-    subprocess.run(["python", "-m", "pytest", "-n", "6", "--log-level=CRITICAL"], check=True)
+    subprocess.run(["python", "-m", "pytest", "-n", "auto", "--log-level=CRITICAL"], check=True)
 
 
 def test_verbose():

tests/integration/test_vectorizers.py

@@ -89,7 +89,7 @@ def embed(text: str):
 
 @pytest.fixture
 def custom_embed_class():
-    class embedder:
+    class MyEmbedder:
         def embed(self, text: str):
             return [1.1, 2.2, 3.3, 4.4]
 
@@ -105,7 +105,7 @@ def embed_many_with_args(self, texts, param=True):
             else:
                 return [[6.0, 5.0, 4.0], [3.0, 2.0, 1.0]]
 
-    return embedder
+    return MyEmbedder
 
 
 def test_vectorizer_embed(vectorizer):