build(deps): drop pytest-cases for pytest 9.1

pyproject.toml

@@ -80,11 +80,11 @@ test = [
     "moto[server]>=5.0,<6",
     "openai-responses>=0.11.4,<1",
     "optuna>=3.0,<5",
-    "pytest>=8.3,<9.1.0",
-    "pytest-asyncio>=1.0,<1.4.0",
+    "pytest>=8.3,<10",
+    "pytest-asyncio>=1.0,<2",
     "pytest-benchmark>=5.1.0",
-    "pytest-cases>=3.8,<4",
     "pytest-env>=1.1,<2",
+    "pytest-lazy-fixtures>=1.4,<2",
     "pytest-rerunfailures>=15.0,<17",
     "ray[default,tune]>=2.40.0,<3",
     "redis>=7.1,<9",

tests/conftest.py

@@ -9,7 +9,6 @@
 
 import pytest
 import pytest_asyncio
-import pytest_cases
 from that_depends import ContextScopes, container_context
 import uvloop
 
@@ -67,16 +66,15 @@ async def DI_teardown() -> _t.AsyncGenerator[None, None]:
         await DI.tear_down()
 
 
-@pytest_cases.fixture
-@pytest_cases.parametrize(zmq_pubsub_proxy=[False, True])
-def zmq_connector_cls(zmq_pubsub_proxy: bool) -> _t.Iterator[_t.Type[ZMQConnector]]:
+@pytest.fixture(params=[False, True], ids=["zmq_pubsub_proxy=False", "zmq_pubsub_proxy=True"])
+def zmq_connector_cls(request: pytest.FixtureRequest) -> _t.Iterator[_t.Type[ZMQConnector]]:
     """Returns the ZMQConnector class with the specified proxy setting.
 
     Patches the env var `PLUGBOARD_FLAGS_ZMQ_PUBSUB_PROXY` to control the proxy setting.
     """
     with patch.dict(
         os.environ,
-        {"PLUGBOARD_FLAGS_ZMQ_PUBSUB_PROXY": str(zmq_pubsub_proxy)},
+        {"PLUGBOARD_FLAGS_ZMQ_PUBSUB_PROXY": str(request.param)},
     ):
         testing_settings = Settings()
         DI.settings.override_sync(testing_settings)

tests/integration/test_channel.py

@@ -6,7 +6,7 @@
 
 from plugboard_schemas.connector import ConnectorMode, ConnectorSpec
 import pytest
-import pytest_cases
+from pytest_lazy_fixtures import lf
 
 from plugboard.connector import (
     Connector,
@@ -23,40 +23,35 @@
 )
 
 
-@pytest_cases.fixture
-@pytest_cases.parametrize(zmq_pubsub_proxy=[True])
-def zmq_connector_cls(zmq_pubsub_proxy: bool) -> _t.Iterator[_t.Type[ZMQConnector]]:
+@pytest.fixture(params=[True], ids=["zmq_pubsub_proxy=True"])
+def zmq_connector_cls(request: pytest.FixtureRequest) -> _t.Iterator[_t.Type[ZMQConnector]]:
     """Returns the ZMQConnector class with the specified proxy setting.
 
     Patches the env var `PLUGBOARD_FLAGS_ZMQ_PUBSUB_PROXY` to control the proxy setting.
     """
     with patch.dict(
         os.environ,
-        {"PLUGBOARD_FLAGS_ZMQ_PUBSUB_PROXY": str(zmq_pubsub_proxy)},
+        {"PLUGBOARD_FLAGS_ZMQ_PUBSUB_PROXY": str(request.param)},
     ):
         testing_settings = Settings()
         DI.settings.override_sync(testing_settings)
         yield ZMQConnector
         DI.settings.reset_override_sync()
 
 
-@pytest_cases.fixture
-@pytest_cases.parametrize("_connector_cls", [RabbitMQConnector, zmq_connector_cls, RedisConnector])
-def connector_cls(_connector_cls: type[Connector]) -> type[Connector]:
+@pytest.fixture(params=[RabbitMQConnector, lf("zmq_connector_cls"), RedisConnector])
+def connector_cls(request: pytest.FixtureRequest) -> type[Connector]:
     """Fixture for `Connector` of various types."""
-    return _connector_cls
+    return request.param
 
 
-@pytest_cases.fixture
-@pytest_cases.parametrize(
-    "_connector_cls_mp", [RabbitMQConnector, zmq_connector_cls, RedisConnector]
-)
-def connector_cls_mp(_connector_cls_mp: type[Connector]) -> type[Connector]:
+@pytest.fixture(params=[RabbitMQConnector, lf("zmq_connector_cls"), RedisConnector])
+def connector_cls_mp(request: pytest.FixtureRequest) -> type[Connector]:
     """Fixture for `Connector` of various types for use in multiprocess context."""
-    return _connector_cls_mp
+    return request.param
 
 
-@pytest_cases.parametrize("connector_cls", [RabbitMQConnector, RedisConnector])
+@pytest.mark.parametrize("connector_cls", [RabbitMQConnector, RedisConnector])
 async def test_channel_broker_url_unset(connector_cls: type[Connector], job_id_ctx: str) -> None:
     """Test that attempting to connect a channel without the broker URL set raises an error."""
     spec = ConnectorSpec(mode=ConnectorMode.PIPELINE, source="test.send", target="test.recv")

tests/integration/test_component_decorator.py

@@ -5,7 +5,6 @@
 import typing as _t
 
 import pytest
-import pytest_cases
 
 from plugboard.component import IOController as IO
 from plugboard.component.utils import component
@@ -83,7 +82,7 @@ async def step(self) -> None:
 
 
 @pytest.mark.asyncio
-@pytest_cases.parametrize(
+@pytest.mark.parametrize(
     "process_cls, connector_cls",
     [
         (LocalProcess, AsyncioConnector),
@@ -127,7 +126,7 @@ async def test_process_with_decorated_components(
 
 
 @pytest.mark.asyncio
-@pytest_cases.parametrize(
+@pytest.mark.parametrize(
     "process_cls, connector_cls",
     [
         (LocalProcess, AsyncioConnector),

tests/integration/test_component_event_handlers.py

@@ -6,14 +6,13 @@
 from pydantic import BaseModel
 import pytest
 import pytest_asyncio
-import pytest_cases
+from pytest_lazy_fixtures import lf
 
 from plugboard.component import Component, IOController
 from plugboard.connector import AsyncioConnector, Connector, ConnectorBuilder
 from plugboard.events import Event
 from plugboard.events.event import StopEvent
 from plugboard.schemas import ConnectorSpec
-from tests.conftest import zmq_connector_cls
 
 
 class EventTypeAData(BaseModel):
@@ -77,11 +76,10 @@ async def event_B_handler(self, evt: EventTypeB) -> None:
         self._event_B_count += evt.data.y
 
 
-@pytest_cases.fixture(scope="function")
-@pytest_cases.parametrize("_connector_cls", [AsyncioConnector, zmq_connector_cls])
-def connector_cls(_connector_cls: _t.Type[Connector]) -> _t.Type[Connector]:
+@pytest.fixture(scope="function", params=[AsyncioConnector, lf("zmq_connector_cls")])
+def connector_cls(request: pytest.FixtureRequest) -> _t.Type[Connector]:
     """Returns a `Connector` class."""
-    return _connector_cls
+    return request.param
 
 
 @pytest.fixture

tests/integration/test_connector_pubsub.py

@@ -5,7 +5,7 @@
 from unittest.mock import patch
 
 import pytest
-import pytest_cases
+from pytest_lazy_fixtures import lf
 
 from plugboard.connector import (
     Connector,
@@ -24,36 +24,34 @@
 )
 
 
-@pytest_cases.fixture
-@pytest_cases.parametrize(zmq_pubsub_proxy=[True])
-def zmq_connector_cls(zmq_pubsub_proxy: bool) -> _t.Iterator[_t.Type[ZMQConnector]]:
+@pytest.fixture(params=[True], ids=["zmq_pubsub_proxy=True"])
+def zmq_connector_cls(request: pytest.FixtureRequest) -> _t.Iterator[_t.Type[ZMQConnector]]:
     """Returns the ZMQConnector class with the specified proxy setting.
 
     Patches the env var `PLUGBOARD_FLAGS_ZMQ_PUBSUB_PROXY` to control the proxy setting.
     """
     with patch.dict(
         os.environ,
-        {"PLUGBOARD_FLAGS_ZMQ_PUBSUB_PROXY": str(zmq_pubsub_proxy)},
+        {"PLUGBOARD_FLAGS_ZMQ_PUBSUB_PROXY": str(request.param)},
     ):
         testing_settings = Settings()
         DI.settings.override_sync(testing_settings)
         yield ZMQConnector
         DI.settings.reset_override_sync()
 
 
-@pytest_cases.fixture
-@pytest_cases.parametrize(_connector_cls=[RabbitMQConnector, zmq_connector_cls, RedisConnector])
-def connector_cls(_connector_cls: type[Connector]) -> type[Connector]:
+@pytest.fixture(params=[RabbitMQConnector, lf("zmq_connector_cls"), RedisConnector])
+def connector_cls(request: pytest.FixtureRequest) -> type[Connector]:
     """Fixture for `Connector` of various types."""
-    return _connector_cls
+    return request.param
 
 
 @pytest.mark.asyncio
-@pytest_cases.parametrize(
-    "connector_cls, num_subscribers, num_messages",
+@pytest.mark.parametrize(
+    "num_subscribers, num_messages",
     [
-        (connector_cls, 1, 100),
-        (connector_cls, 10, 100),
+        (1, 100),
+        (10, 100),
     ],
 )
 async def test_pubsub_channel_single_publisher(
@@ -68,11 +66,11 @@ async def test_pubsub_channel_single_publisher(
 
 
 @pytest.mark.asyncio
-@pytest_cases.parametrize(
-    "connector_cls, num_publishers, num_subscribers, num_messages",
+@pytest.mark.parametrize(
+    "num_publishers, num_subscribers, num_messages",
     [
-        (connector_cls, 10, 1, 100),
-        (connector_cls, 10, 10, 100),
+        (10, 1, 100),
+        (10, 10, 100),
     ],
 )
 async def test_pubsub_channel_multiple_publishers(
@@ -93,11 +91,11 @@ async def test_pubsub_channel_multiple_publishers(
 
 
 @pytest.mark.asyncio
-@pytest_cases.parametrize(
-    "connector_cls, num_topics, num_publishers, num_subscribers, num_messages",
+@pytest.mark.parametrize(
+    "num_topics, num_publishers, num_subscribers, num_messages",
     [
-        (connector_cls, 3, 10, 1, 100),
-        (connector_cls, 3, 10, 10, 100),
+        (3, 10, 1, 100),
+        (3, 10, 10, 100),
     ],
 )
 async def test_pubsub_channel_multiple_topics_and_publishers(

tests/integration/test_process_stop_cancel.py

@@ -7,7 +7,7 @@
 import typing as _t
 
 import pytest
-import pytest_cases
+from pytest_lazy_fixtures import lf
 
 from plugboard.component import Component, IOController as IO
 from plugboard.connector import (
@@ -20,7 +20,7 @@
 from plugboard.events import StopEvent
 from plugboard.process import LocalProcess, Process, RayProcess
 from plugboard.schemas import ConnectorSpec, Status
-from tests.conftest import ComponentTestHelper, zmq_connector_cls
+from tests.conftest import ComponentTestHelper
 
 
 STOP_TOLERANCE = 3
@@ -57,14 +57,14 @@ async def step(self) -> None:
 
 
 @pytest.mark.asyncio
-@pytest_cases.parametrize(
+@pytest.mark.parametrize(
     "process_cls, connector_cls",
     [
         (LocalProcess, AsyncioConnector),
-        (LocalProcess, zmq_connector_cls),
+        (LocalProcess, lf("zmq_connector_cls")),
         (LocalProcess, RabbitMQConnector),
         # (RayProcess, RayConnector),  # TODO : Pubsub/StopEvent unsupported. See https://github.com/plugboard-dev/plugboard/issues/101.
-        (RayProcess, zmq_connector_cls),
+        (RayProcess, lf("zmq_connector_cls")),
         (RayProcess, RabbitMQConnector),
     ],
 )
@@ -137,7 +137,7 @@ async def stop_after() -> None:
 
 
 @pytest.mark.asyncio
-@pytest_cases.parametrize(
+@pytest.mark.parametrize(
     "process_cls, connector_cls",
     [
         (LocalProcess, AsyncioConnector),

tests/integration/test_process_with_components_run.py

@@ -10,7 +10,7 @@
 from aiofile import async_open
 from pydantic import BaseModel
 import pytest
-import pytest_cases
+from pytest_lazy_fixtures import lf
 
 from plugboard.component import IOController as IO
 from plugboard.component.component import IO_READ_TIMEOUT_SECONDS
@@ -26,7 +26,7 @@
 from plugboard.library import FileWriter
 from plugboard.process import LocalProcess, Process, RayProcess
 from plugboard.schemas import ConnectorSpec, Status
-from tests.conftest import ComponentTestHelper, zmq_connector_cls
+from tests.conftest import ComponentTestHelper
 
 
 class A(ComponentTestHelper):
@@ -84,14 +84,14 @@ def tempfile_path() -> _t.Generator[Path, None, None]:
 
 
 @pytest.mark.asyncio
-@pytest_cases.parametrize(
+@pytest.mark.parametrize(
     "process_cls, connector_cls",
     [
         (LocalProcess, AsyncioConnector),
-        (LocalProcess, zmq_connector_cls),
+        (LocalProcess, lf("zmq_connector_cls")),
         (LocalProcess, RabbitMQConnector),
         (RayProcess, RayConnector),
-        (RayProcess, zmq_connector_cls),
+        (RayProcess, lf("zmq_connector_cls")),
         (RayProcess, RabbitMQConnector),
     ],
 )
@@ -200,7 +200,7 @@ async def _status_check(self) -> None:
 
 
 @pytest.mark.asyncio
-@pytest_cases.parametrize(
+@pytest.mark.parametrize(
     "process_cls, connector_cls",
     [
         (LocalProcess, AsyncioConnector),
@@ -259,7 +259,7 @@ async def step(self) -> None:
 
 
 @pytest.mark.asyncio
-@pytest_cases.parametrize(
+@pytest.mark.parametrize(
     "process_cls, connector_cls",
     [
         (LocalProcess, AsyncioConnector),
@@ -420,7 +420,7 @@ async def handle_action(self, evt: ActionEvent) -> None:
 
 
 @pytest.mark.asyncio
-@pytest_cases.parametrize(
+@pytest.mark.parametrize(
     "process_cls, connector_cls",
     [
         (LocalProcess, AsyncioConnector),
@@ -659,7 +659,7 @@ async def handle_message(self, event: MessageEvent) -> None:
 
 
 @pytest.mark.asyncio
-@pytest_cases.parametrize(
+@pytest.mark.parametrize(
     "process_cls, connector_cls",
     [
         (LocalProcess, AsyncioConnector),

tests/integration/test_state_backend_multiprocess.py

@@ -4,7 +4,6 @@
 
 import pytest
 import pytest_asyncio
-import pytest_cases
 from ray.util.multiprocessing import Pool
 import uvloop
 
@@ -185,7 +184,7 @@ async def _inner() -> None:
         assert state_data_conn["times_upserted"] == 2
 
 
-@pytest_cases.parametrize("setup_backend", [setup_SqliteStateBackend, setup_PostgresStateBackend])
+@pytest.mark.parametrize("setup_backend", [setup_SqliteStateBackend, setup_PostgresStateBackend])
 @pytest.mark.asyncio
 async def test_no_process_found_errors(
     setup_backend: _t.Callable[[], _t.ContextManager[StateBackend]],