clean up imports in scenario ops unit tests

Author: sid-rl

tests/sdk/test_async_ops.py

@@ -24,6 +24,7 @@
     AsyncAgent,
     AsyncDevbox,
     AsyncScorer,
+    AsyncScenario,
     AsyncSnapshot,
     AsyncBlueprint,
     AsyncStorageObject,
@@ -33,6 +34,7 @@
     AsyncDevboxOps,
     AsyncScorerOps,
     AsyncRunloopSDK,
+    AsyncScenarioOps,
     AsyncSnapshotOps,
     AsyncBlueprintOps,
     AsyncStorageObjectOps,
@@ -1122,8 +1124,6 @@ class TestAsyncScenarioOps:
 
     def test_from_id(self, mock_async_client: AsyncMock) -> None:
         """Test from_id method."""
-        from runloop_api_client.sdk import AsyncScenario
-        from runloop_api_client.sdk.async_ import AsyncScenarioOps
 
         ops = AsyncScenarioOps(mock_async_client)
         scenario = ops.from_id("scn_123")
@@ -1134,7 +1134,6 @@ def test_from_id(self, mock_async_client: AsyncMock) -> None:
     @pytest.mark.asyncio
     async def test_list_empty(self, mock_async_client: AsyncMock) -> None:
         """Test list method with empty results."""
-        from runloop_api_client.sdk.async_ import AsyncScenarioOps
 
         async def async_iter():
             return
@@ -1151,8 +1150,6 @@ async def async_iter():
     @pytest.mark.asyncio
     async def test_list_single(self, mock_async_client: AsyncMock, scenario_view: MockScenarioView) -> None:
         """Test list method with single result."""
-        from runloop_api_client.sdk import AsyncScenario
-        from runloop_api_client.sdk.async_ import AsyncScenarioOps
 
         async def async_iter():
             yield scenario_view
@@ -1170,8 +1167,6 @@ async def async_iter():
     @pytest.mark.asyncio
     async def test_list_multiple(self, mock_async_client: AsyncMock) -> None:
         """Test list method with multiple results."""
-        from runloop_api_client.sdk import AsyncScenario
-        from runloop_api_client.sdk.async_ import AsyncScenarioOps
 
         scenario_view1 = MockScenarioView(id="scn_001", name="scenario-1")
         scenario_view2 = MockScenarioView(id="scn_002", name="scenario-2")

tests/sdk/test_ops.py

@@ -20,12 +20,13 @@
     MockBlueprintView,
     create_mock_httpx_response,
 )
-from runloop_api_client.sdk import Agent, Devbox, Scorer, Snapshot, Blueprint, StorageObject
+from runloop_api_client.sdk import Agent, Devbox, Scorer, Scenario, Snapshot, Blueprint, StorageObject
 from runloop_api_client.sdk.sync import (
     AgentOps,
     DevboxOps,
     ScorerOps,
     RunloopSDK,
+    ScenarioOps,
     SnapshotOps,
     BlueprintOps,
     StorageObjectOps,
@@ -1015,8 +1016,6 @@ class TestScenarioOps:
 
     def test_from_id(self, mock_client: Mock) -> None:
         """Test from_id method."""
-        from runloop_api_client.sdk import Scenario
-        from runloop_api_client.sdk.sync import ScenarioOps
 
         ops = ScenarioOps(mock_client)
         scenario = ops.from_id("scn_123")
@@ -1026,7 +1025,6 @@ def test_from_id(self, mock_client: Mock) -> None:
 
     def test_list_empty(self, mock_client: Mock) -> None:
         """Test list method with empty results."""
-        from runloop_api_client.sdk.sync import ScenarioOps
 
         mock_client.scenarios.list.return_value = []
 
@@ -1038,8 +1036,6 @@ def test_list_empty(self, mock_client: Mock) -> None:
 
     def test_list_single(self, mock_client: Mock, scenario_view: MockScenarioView) -> None:
         """Test list method with single result."""
-        from runloop_api_client.sdk import Scenario
-        from runloop_api_client.sdk.sync import ScenarioOps
 
         mock_client.scenarios.list.return_value = [scenario_view]
 
@@ -1053,8 +1049,6 @@ def test_list_single(self, mock_client: Mock, scenario_view: MockScenarioView) -
 
     def test_list_multiple(self, mock_client: Mock) -> None:
         """Test list method with multiple results."""
-        from runloop_api_client.sdk import Scenario
-        from runloop_api_client.sdk.sync import ScenarioOps
 
         scenario_view1 = MockScenarioView(id="scn_001", name="scenario-1")
         scenario_view2 = MockScenarioView(id="scn_002", name="scenario-2")