estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

chore(api/tests): apply ruff reformat #7590 (#7591)

Browse Source 
Co-authored-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
Bowen Liang
2024-08-23 23:52:25 +08:00
committed by GitHub
parent 2da63654e5
commit b035c02f78
155 changed files with 4279 additions and 5925 deletions
Download Patch File Download Diff File Expand all files Collapse all files

278
api/tests/integration_tests/workflow/nodes/test_code.py
View File

@@ -9,137 +9,134 @@ from core.workflow.nodes.code.code_node import CodeNode
from models.workflow import WorkflowNodeExecutionStatus
from tests.integration_tests.workflow.nodes.__mock.code_executor import setup_code_executor_mock
CODE_MAX_STRING_LENGTH = int(getenv('CODE_MAX_STRING_LENGTH', '10000'))
CODE_MAX_STRING_LENGTH = int(getenv("CODE_MAX_STRING_LENGTH", "10000"))
@pytest.mark.parametrize('setup_code_executor_mock', [['none']], indirect=True)
@pytest.mark.parametrize("setup_code_executor_mock", [["none"]], indirect=True)
def test_execute_code(setup_code_executor_mock):
code = '''
code = """
def main(args1: int, args2: int) -> dict:
return {
"result": args1 + args2,
}
'''
"""
# trim first 4 spaces at the beginning of each line
code = '\n'.join([line[4:] for line in code.split('\n')])
code = "\n".join([line[4:] for line in code.split("\n")])
node = CodeNode(
tenant_id='1',
app_id='1',
workflow_id='1',
user_id='1',
tenant_id="1",
app_id="1",
workflow_id="1",
user_id="1",
user_from=UserFrom.ACCOUNT,
invoke_from=InvokeFrom.WEB_APP,
config={
'id': '1',
'data': {
'outputs': {
'result': {
'type': 'number',
"id": "1",
"data": {
"outputs": {
"result": {
"type": "number",
},
},
'title': '123',
'variables': [
"title": "123",
"variables": [
{
'variable': 'args1',
'value_selector': ['1', '123', 'args1'],
"variable": "args1",
"value_selector": ["1", "123", "args1"],
},
{
'variable': 'args2',
'value_selector': ['1', '123', 'args2']
}
{"variable": "args2", "value_selector": ["1", "123", "args2"]},
],
'answer': '123',
'code_language': 'python3',
'code': code
}
}
"answer": "123",
"code_language": "python3",
"code": code,
},
},
)
# construct variable pool
pool = VariablePool(system_variables={}, user_inputs={}, environment_variables=[])
pool.add(['1', '123', 'args1'], 1)
pool.add(['1', '123', 'args2'], 2)
pool.add(["1", "123", "args1"], 1)
pool.add(["1", "123", "args2"], 2)
# execute node
result = node.run(pool)
assert result.status == WorkflowNodeExecutionStatus.SUCCEEDED
assert result.outputs['result'] == 3
assert result.outputs["result"] == 3
assert result.error is None
@pytest.mark.parametrize('setup_code_executor_mock', [['none']], indirect=True)
@pytest.mark.parametrize("setup_code_executor_mock", [["none"]], indirect=True)
def test_execute_code_output_validator(setup_code_executor_mock):
code = '''
code = """
def main(args1: int, args2: int) -> dict:
return {
"result": args1 + args2,
}
'''
"""
# trim first 4 spaces at the beginning of each line
code = '\n'.join([line[4:] for line in code.split('\n')])
code = "\n".join([line[4:] for line in code.split("\n")])
node = CodeNode(
tenant_id='1',
app_id='1',
workflow_id='1',
user_id='1',
tenant_id="1",
app_id="1",
workflow_id="1",
user_id="1",
user_from=UserFrom.ACCOUNT,
invoke_from=InvokeFrom.WEB_APP,
config={
'id': '1',
'data': {
"id": "1",
"data": {
"outputs": {
"result": {
"type": "string",
},
},
'title': '123',
'variables': [
"title": "123",
"variables": [
{
'variable': 'args1',
'value_selector': ['1', '123', 'args1'],
"variable": "args1",
"value_selector": ["1", "123", "args1"],
},
{
'variable': 'args2',
'value_selector': ['1', '123', 'args2']
}
{"variable": "args2", "value_selector": ["1", "123", "args2"]},
],
'answer': '123',
'code_language': 'python3',
'code': code
}
}
"answer": "123",
"code_language": "python3",
"code": code,
},
},
)
# construct variable pool
pool = VariablePool(system_variables={}, user_inputs={}, environment_variables=[])
pool.add(['1', '123', 'args1'], 1)
pool.add(['1', '123', 'args2'], 2)
pool.add(["1", "123", "args1"], 1)
pool.add(["1", "123", "args2"], 2)
# execute node
result = node.run(pool)
assert result.status == WorkflowNodeExecutionStatus.FAILED
assert result.error == 'Output variable `result` must be a string'
assert result.error == "Output variable `result` must be a string"
def test_execute_code_output_validator_depth():
code = '''
code = """
def main(args1: int, args2: int) -> dict:
return {
"result": {
"result": args1 + args2,
}
}
'''
"""
# trim first 4 spaces at the beginning of each line
code = '\n'.join([line[4:] for line in code.split('\n')])
code = "\n".join([line[4:] for line in code.split("\n")])
node = CodeNode(
tenant_id='1',
app_id='1',
workflow_id='1',
user_id='1',
tenant_id="1",
app_id="1",
workflow_id="1",
user_id="1",
user_from=UserFrom.ACCOUNT,
invoke_from=InvokeFrom.WEB_APP,
config={
'id': '1',
'data': {
"id": "1",
"data": {
"outputs": {
"string_validator": {
"type": "string",
@@ -168,29 +165,26 @@ def test_execute_code_output_validator_depth():
"depth": {
"type": "number",
}
}
},
}
}
}
}
},
},
},
},
},
'title': '123',
'variables': [
"title": "123",
"variables": [
{
'variable': 'args1',
'value_selector': ['1', '123', 'args1'],
"variable": "args1",
"value_selector": ["1", "123", "args1"],
},
{
'variable': 'args2',
'value_selector': ['1', '123', 'args2']
}
{"variable": "args2", "value_selector": ["1", "123", "args2"]},
],
'answer': '123',
'code_language': 'python3',
'code': code
}
}
"answer": "123",
"code_language": "python3",
"code": code,
},
},
)
# construct result
@@ -199,14 +193,7 @@ def test_execute_code_output_validator_depth():
"string_validator": "1",
"number_array_validator": [1, 2, 3, 3.333],
"string_array_validator": ["1", "2", "3"],
"object_validator": {
"result": 1,
"depth": {
"depth": {
"depth": 1
}
}
}
"object_validator": {"result": 1, "depth": {"depth": {"depth": 1}}},
}
# validate
@@ -218,14 +205,7 @@ def test_execute_code_output_validator_depth():
"string_validator": 1,
"number_array_validator": ["1", "2", "3", "3.333"],
"string_array_validator": [1, 2, 3],
"object_validator": {
"result": "1",
"depth": {
"depth": {
"depth": "1"
}
}
}
"object_validator": {"result": "1", "depth": {"depth": {"depth": "1"}}},
}
# validate
@@ -238,34 +218,20 @@ def test_execute_code_output_validator_depth():
"string_validator": (CODE_MAX_STRING_LENGTH + 1) * "1",
"number_array_validator": [1, 2, 3, 3.333],
"string_array_validator": ["1", "2", "3"],
"object_validator": {
"result": 1,
"depth": {
"depth": {
"depth": 1
}
}
}
"object_validator": {"result": 1, "depth": {"depth": {"depth": 1}}},
}
# validate
with pytest.raises(ValueError):
node._transform_result(result, node.node_data.outputs)
# construct result
result = {
"number_validator": 1,
"string_validator": "1",
"number_array_validator": [1, 2, 3, 3.333] * 2000,
"string_array_validator": ["1", "2", "3"],
"object_validator": {
"result": 1,
"depth": {
"depth": {
"depth": 1
}
}
}
"object_validator": {"result": 1, "depth": {"depth": {"depth": 1}}},
}
# validate
@@ -274,58 +240,59 @@ def test_execute_code_output_validator_depth():
def test_execute_code_output_object_list():
code = '''
code = """
def main(args1: int, args2: int) -> dict:
return {
"result": {
"result": args1 + args2,
}
}
'''
"""
# trim first 4 spaces at the beginning of each line
code = '\n'.join([line[4:] for line in code.split('\n')])
code = "\n".join([line[4:] for line in code.split("\n")])
node = CodeNode(
tenant_id='1',
app_id='1',
workflow_id='1',
user_id='1',
tenant_id="1",
app_id="1",
workflow_id="1",
user_id="1",
invoke_from=InvokeFrom.WEB_APP,
user_from=UserFrom.ACCOUNT,
config={
'id': '1',
'data': {
"id": "1",
"data": {
"outputs": {
"object_list": {
"type": "array[object]",
},
},
'title': '123',
'variables': [
"title": "123",
"variables": [
{
'variable': 'args1',
'value_selector': ['1', '123', 'args1'],
"variable": "args1",
"value_selector": ["1", "123", "args1"],
},
{
'variable': 'args2',
'value_selector': ['1', '123', 'args2']
}
{"variable": "args2", "value_selector": ["1", "123", "args2"]},
],
'answer': '123',
'code_language': 'python3',
'code': code
}
}
"answer": "123",
"code_language": "python3",
"code": code,
},
},
)
# construct result
result = {
"object_list": [{
"result": 1,
}, {
"result": 2,
}, {
"result": [1, 2, 3],
}]
"object_list": [
{
"result": 1,
},
{
"result": 2,
},
{
"result": [1, 2, 3],
},
]
}
# validate
@@ -333,13 +300,18 @@ def test_execute_code_output_object_list():
# construct result
result = {
"object_list": [{
"result": 1,
}, {
"result": 2,
}, {
"result": [1, 2, 3],
}, 1]
"object_list": [
{
"result": 1,
},
{
"result": 2,
},
{
"result": [1, 2, 3],
},
1,
]
}
# validate