update tests

tests/plugins/orderer/_utils.py

@@ -1,9 +1,12 @@
+import random
 from argparse import Namespace
+from contextlib import contextmanager
 from pathlib import Path
 
 import pytest
 
-from vedro.core import ArgumentParser, ConfigType, Dispatcher
+from vedro import Scenario
+from vedro.core import ArgumentParser, ConfigType, Dispatcher, VirtualScenario
 from vedro.events import ArgParsedEvent, ArgParseEvent, ConfigLoadedEvent
 from vedro.plugins.orderer import Orderer, OrdererPlugin
 
@@ -20,6 +23,21 @@ def orderer(dispatcher: Dispatcher) -> OrdererPlugin:
     return orderer
 
 
+@contextmanager
+def seeded(seed: str):
+    state = random.getstate()
+    random.seed(seed)
+    yield random.seed(seed)
+    random.setstate(state)
+
+
+def make_vscenario(path: str) -> VirtualScenario:
+    class _Scenario(Scenario):
+        __file__ = Path(path).absolute()
+
+    return VirtualScenario(_Scenario, steps=[])
+
+
 async def fire_config_loaded_event(dispatcher: Dispatcher, config: ConfigType) -> None:
     config_loaded_event = ConfigLoadedEvent(Path(), config)
     await dispatcher.fire(config_loaded_event)

tests/plugins/orderer/test_random_scenario_orderer.py

@@ -1,35 +1,16 @@
-import random
-from contextlib import contextmanager
-from pathlib import Path
-
 import pytest
 from baby_steps import given, then, when
 
-from vedro import Scenario
-from vedro.core import VirtualScenario
 from vedro.plugins.orderer import RandomOrderer
 
+from ._utils import make_vscenario, seeded
+
 
 @pytest.fixture()
-def orderer():
+def orderer() -> RandomOrderer:
     return RandomOrderer()
 
 
-@contextmanager
-def seeded(seed: str):
-    state = random.getstate()
-    random.seed(seed)
-    yield
-    random.setstate(state)
-
-
-def make_vscenario(path: str) -> VirtualScenario:
-    class _Scenario(Scenario):
-        __file__ = Path(path).absolute()
-
-    return VirtualScenario(_Scenario, steps=[])
-
-
 @pytest.mark.asyncio
 async def test_sort_no_scenarios(*, orderer: RandomOrderer):
     with given:

tests/plugins/orderer/test_reversed_scenario_orderer.py

@@ -1,25 +1,16 @@
-from pathlib import Path
-
 import pytest
 from baby_steps import given, then, when
 
-from vedro import Scenario
-from vedro.core import VirtualScenario
 from vedro.plugins.orderer import ReversedOrderer
 
+from ._utils import make_vscenario
+
 
 @pytest.fixture()
-def orderer():
+def orderer() -> ReversedOrderer:
     return ReversedOrderer()
 
 
-def make_vscenario(path: str) -> VirtualScenario:
-    class _Scenario(Scenario):
-        __file__ = Path(path).absolute()
-
-    return VirtualScenario(_Scenario, steps=[])
-
-
 @pytest.mark.asyncio
 async def test_sort_no_scenarios(*, orderer: ReversedOrderer):
     with given:

tests/plugins/orderer/test_stable_scenario_orderer.py

@@ -1,25 +1,16 @@
-from pathlib import Path
-
 import pytest
 from baby_steps import given, then, when
 
-from vedro import Scenario
-from vedro.core import VirtualScenario
 from vedro.plugins.orderer import StableScenarioOrderer
 
+from ._utils import make_vscenario
+
 
 @pytest.fixture()
-def orderer():
+def orderer() -> StableScenarioOrderer:
     return StableScenarioOrderer()
 
 
-def make_vscenario(path: str) -> VirtualScenario:
-    class _Scenario(Scenario):
-        __file__ = Path(path).absolute()
-
-    return VirtualScenario(_Scenario, steps=[])
-
-
 @pytest.mark.asyncio
 async def test_sort_no_scenarios(*, orderer: StableScenarioOrderer):
     with given: