Spaces:
Runtime error
Runtime error
File size: 8,524 Bytes
e331e72 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 |
# Copyright (c) 2024 Microsoft Corporation.
# Licensed under the MIT License
import unittest
import pytest
from graphrag.index.config import PipelineWorkflowReference
from graphrag.index.errors import UnknownWorkflowError
from graphrag.index.workflows.load import create_workflow, load_workflows
from .helpers import mock_verbs, mock_workflows
class TestCreateWorkflow(unittest.TestCase):
def test_workflow_with_steps_should_not_fail(self):
create_workflow(
"workflow_with_steps",
[
{
"verb": "mock_verb",
"args": {
"column": "test",
},
}
],
config=None,
additional_verbs=mock_verbs,
)
def test_non_existent_workflow_without_steps_should_crash(self):
# since we don't have a workflow named "test", and the user didn't provide any steps, we should crash
# since we don't know what to do
with pytest.raises(UnknownWorkflowError):
create_workflow("test", None, config=None, additional_verbs=mock_verbs)
def test_existing_workflow_should_not_crash(self):
create_workflow(
"mock_workflow",
None,
config=None,
additional_verbs=mock_verbs,
additional_workflows=mock_workflows,
)
class TestLoadWorkflows(unittest.TestCase):
def test_non_existent_workflow_should_crash(self):
with pytest.raises(UnknownWorkflowError):
load_workflows(
[
PipelineWorkflowReference(
name="some_workflow_that_does_not_exist",
config=None,
)
],
additional_workflows=mock_workflows,
additional_verbs=mock_verbs,
)
def test_single_workflow_should_not_crash(self):
load_workflows(
[
PipelineWorkflowReference(
name="mock_workflow",
config=None,
)
],
additional_workflows=mock_workflows,
additional_verbs=mock_verbs,
)
def test_multiple_workflows_should_not_crash(self):
load_workflows(
[
PipelineWorkflowReference(
name="mock_workflow",
config=None,
),
PipelineWorkflowReference(
name="mock_workflow_2",
config=None,
),
],
# the two above are in the "mock_workflows" list
additional_workflows=mock_workflows,
additional_verbs=mock_verbs,
)
def test_two_interdependent_workflows_should_provide_correct_order(self):
ordered_workflows, _deps = load_workflows(
[
PipelineWorkflowReference(
name="interdependent_workflow_1",
steps=[
{
"verb": "mock_verb",
"args": {
"column": "test",
},
"input": {
"source": "workflow:interdependent_workflow_2"
}, # This one is dependent on the second one, so when it comes out of load_workflows, it should be first
}
],
),
PipelineWorkflowReference(
name="interdependent_workflow_2",
steps=[
{
"verb": "mock_verb",
"args": {
"column": "test",
},
}
],
),
],
# the two above are in the "mock_workflows" list
additional_workflows=mock_workflows,
additional_verbs=mock_verbs,
)
# two should only come out
assert len(ordered_workflows) == 2
assert ordered_workflows[0].workflow.name == "interdependent_workflow_2"
assert ordered_workflows[1].workflow.name == "interdependent_workflow_1"
def test_three_interdependent_workflows_should_provide_correct_order(self):
ordered_workflows, _deps = load_workflows(
[
PipelineWorkflowReference(
name="interdependent_workflow_3",
steps=[
{
"verb": "mock_verb",
"args": {
"column": "test",
},
}
],
),
PipelineWorkflowReference(
name="interdependent_workflow_1",
steps=[
{
"verb": "mock_verb",
"args": {
"column": "test",
},
"input": {"source": "workflow:interdependent_workflow_2"},
}
],
),
PipelineWorkflowReference(
name="interdependent_workflow_2",
steps=[
{
"verb": "mock_verb",
"args": {
"column": "test",
},
"input": {"source": "workflow:interdependent_workflow_3"},
}
],
),
],
# the two above are in the "mock_workflows" list
additional_workflows=mock_workflows,
additional_verbs=mock_verbs,
)
order = [
"interdependent_workflow_3",
"interdependent_workflow_2",
"interdependent_workflow_1",
]
assert [x.workflow.name for x in ordered_workflows] == order
def test_two_workflows_dependent_on_another_single_workflow_should_provide_correct_order(
self,
):
ordered_workflows, _deps = load_workflows(
[
# Workflows 1 and 2 are dependent on 3, so 3 should come out first
PipelineWorkflowReference(
name="interdependent_workflow_3",
steps=[
{
"verb": "mock_verb",
"args": {
"column": "test",
},
}
],
),
PipelineWorkflowReference(
name="interdependent_workflow_1",
steps=[
{
"verb": "mock_verb",
"args": {
"column": "test",
},
"input": {"source": "workflow:interdependent_workflow_3"},
}
],
),
PipelineWorkflowReference(
name="interdependent_workflow_2",
steps=[
{
"verb": "mock_verb",
"args": {
"column": "test",
},
"input": {"source": "workflow:interdependent_workflow_3"},
}
],
),
],
# the two above are in the "mock_workflows" list
additional_workflows=mock_workflows,
additional_verbs=mock_verbs,
)
assert len(ordered_workflows) == 3
assert ordered_workflows[0].workflow.name == "interdependent_workflow_3"
# The order of the other two doesn't matter, but they need to be there
assert ordered_workflows[1].workflow.name in [
"interdependent_workflow_1",
"interdependent_workflow_2",
]
assert ordered_workflows[2].workflow.name in [
"interdependent_workflow_1",
"interdependent_workflow_2",
]
|