Skip to content

Commit 1ef82d1

Browse files
test(python): group local input regressions before fixtures
1 parent 6dde9ae commit 1ef82d1

1 file changed

Lines changed: 29 additions & 28 deletions

File tree

‎python/tests/unit/processes/local_runtime/test_local_process.py‎

Lines changed: 29 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@
1818
)
1919
from semantic_kernel.processes.kernel_process.kernel_process_state import KernelProcessState
2020
from semantic_kernel.processes.kernel_process.kernel_process_step import KernelProcessStep
21-
from semantic_kernel.processes.kernel_process.kernel_process_step_context import KernelProcessStepContext
2221
from semantic_kernel.processes.kernel_process.kernel_process_step_info import KernelProcessStepInfo
2322
from semantic_kernel.processes.local_runtime.local_kernel_process import start
2423
from semantic_kernel.processes.local_runtime.local_message import LocalMessage
@@ -336,36 +335,11 @@ async def test_handle_message_with_valid_event_id(mock_process_with_output_edges
336335
END_PROCESS_ID = "END"
337336

338337

339-
@pytest.fixture
340-
def mock_process_with_steps(mock_process, mock_kernel, build_model):
341-
"""Fixture to create a LocalProcess with mock steps."""
342-
local_process = LocalProcess(process=mock_process, kernel=mock_kernel)
343-
344-
# Create real LocalStep instances
345-
state_1 = MagicMock(spec=KernelProcessState)
346-
state_1.name = "step_1_state"
347-
state_1.id = "step_1_id"
348-
349-
step_info_1 = MagicMock(spec=KernelProcessStepInfo)
350-
step_info_1.state = state_1
351-
352-
state_2 = MagicMock(spec=KernelProcessState)
353-
state_2.name = "step_2_state"
354-
state_2.id = "step_2_id"
355-
356-
step_info_2 = MagicMock(spec=KernelProcessStepInfo)
357-
step_info_2.state = state_2
358-
359-
step_1 = LocalStep(step_info=step_info_1, kernel=mock_kernel, parent_process_id="test_process")
360-
step_2 = LocalStep(step_info=step_info_2, kernel=mock_kernel, parent_process_id="test_process")
361-
362-
local_process.steps = [step_1, step_2]
363-
return local_process
364-
365-
366338
@pytest.mark.parametrize("defer_second_b", [False, True])
367339
@pytest.mark.parametrize("fail_first", [False, True])
368340
async def test_messages_to_same_step_do_not_reuse_inputs_during_invocation(defer_second_b, fail_first):
341+
from semantic_kernel.processes.kernel_process.kernel_process_step_context import KernelProcessStepContext
342+
369343
pairs = []
370344

371345
class Producer(KernelProcessStep):
@@ -403,3 +377,30 @@ async def relay(self, value: int) -> int:
403377
pass
404378

405379
assert pairs == [(1, 1), (2, 2)]
380+
381+
382+
@pytest.fixture
383+
def mock_process_with_steps(mock_process, mock_kernel, build_model):
384+
"""Fixture to create a LocalProcess with mock steps."""
385+
local_process = LocalProcess(process=mock_process, kernel=mock_kernel)
386+
387+
# Create real LocalStep instances
388+
state_1 = MagicMock(spec=KernelProcessState)
389+
state_1.name = "step_1_state"
390+
state_1.id = "step_1_id"
391+
392+
step_info_1 = MagicMock(spec=KernelProcessStepInfo)
393+
step_info_1.state = state_1
394+
395+
state_2 = MagicMock(spec=KernelProcessState)
396+
state_2.name = "step_2_state"
397+
state_2.id = "step_2_id"
398+
399+
step_info_2 = MagicMock(spec=KernelProcessStepInfo)
400+
step_info_2.state = state_2
401+
402+
step_1 = LocalStep(step_info=step_info_1, kernel=mock_kernel, parent_process_id="test_process")
403+
step_2 = LocalStep(step_info=step_info_2, kernel=mock_kernel, parent_process_id="test_process")
404+
405+
local_process.steps = [step_1, step_2]
406+
return local_process

0 commit comments

Comments
 (0)