From bcf0b88a7c5109b9534aa20ea8cb672d954efc05 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 14 Jun 2023 07:21:49 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=AA=20test(process.py):=20add=20tests?= =?UTF-8?q?=20for=20process=5Ftweaks=20function=20This=20commit=20adds=20t?= =?UTF-8?q?ests=20for=20the=20process=5Ftweaks=20function=20in=20the=20lan?= =?UTF-8?q?gflow.processing.process=20module.=20The=20tests=20cover=20the?= =?UTF-8?q?=20following=20scenarios:=20-=20No=20tweaks=20are=20applied=20t?= =?UTF-8?q?o=20the=20graph=20data=20-=20A=20single=20tweak=20is=20applied?= =?UTF-8?q?=20to=20the=20graph=20data=20-=20Multiple=20tweaks=20are=20appl?= =?UTF-8?q?ied=20to=20the=20graph=20data=20-=20A=20tweak=20is=20applied=20?= =?UTF-8?q?to=20a=20non-existent=20parameter=20in=20the=20graph=20data=20T?= =?UTF-8?q?he=20tests=20ensure=20that=20the=20function=20processes=20the?= =?UTF-8?q?=20tweaks=20correctly=20and=20returns=20the=20expected=20result?= =?UTF-8?q?.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_process.py | 196 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 196 insertions(+) create mode 100644 tests/test_process.py diff --git a/tests/test_process.py b/tests/test_process.py new file mode 100644 index 000000000..2d0c349ce --- /dev/null +++ b/tests/test_process.py @@ -0,0 +1,196 @@ +from langflow.processing.process import process_tweaks + + +def test_no_tweaks(): + graph_data = { + "data": { + "nodes": [ + { + "id": "node1", + "data": { + "node": { + "template": { + "param1": {"value": 1}, + "param2": {"value": 2}, + } + } + }, + }, + { + "id": "node2", + "data": { + "node": { + "template": { + "param1": {"value": 3}, + "param2": {"value": 4}, + } + } + }, + }, + ] + } + } + tweaks = {} + result = process_tweaks(graph_data, tweaks) + assert result == graph_data + + +def test_single_tweak(): + graph_data = { + "data": { + "nodes": [ + { + "id": "node1", + "data": { + "node": { + "template": { + "param1": {"value": 1}, + "param2": {"value": 2}, + } + } + }, + }, + { + "id": "node2", + "data": { + "node": { + "template": { + "param1": {"value": 3}, + "param2": {"value": 4}, + } + } + }, + }, + ] + } + } + tweaks = {"node1": {"param1": 5}} + expected_result = { + "data": { + "nodes": [ + { + "id": "node1", + "data": { + "node": { + "template": { + "param1": {"value": 5}, + "param2": {"value": 2}, + } + } + }, + }, + { + "id": "node2", + "data": { + "node": { + "template": { + "param1": {"value": 3}, + "param2": {"value": 4}, + } + } + }, + }, + ] + } + } + result = process_tweaks(graph_data, tweaks) + assert result == expected_result + + +def test_multiple_tweaks(): + graph_data = { + "data": { + "nodes": [ + { + "id": "node1", + "data": { + "node": { + "template": { + "param1": {"value": 1}, + "param2": {"value": 2}, + } + } + }, + }, + { + "id": "node2", + "data": { + "node": { + "template": { + "param1": {"value": 3}, + "param2": {"value": 4}, + } + } + }, + }, + ] + } + } + tweaks = { + "node1": {"param1": 5, "param2": 6}, + "node2": {"param1": 7}, + } + expected_result = { + "data": { + "nodes": [ + { + "id": "node1", + "data": { + "node": { + "template": { + "param1": {"value": 5}, + "param2": {"value": 6}, + } + } + }, + }, + { + "id": "node2", + "data": { + "node": { + "template": { + "param1": {"value": 7}, + "param2": {"value": 4}, + } + } + }, + }, + ] + } + } + result = process_tweaks(graph_data, tweaks) + assert result == expected_result + + +def test_tweak_not_in_template(): + graph_data = { + "data": { + "nodes": [ + { + "id": "node1", + "data": { + "node": { + "template": { + "param1": {"value": 1}, + "param2": {"value": 2}, + } + } + }, + }, + { + "id": "node2", + "data": { + "node": { + "template": { + "param1": {"value": 3}, + "param2": {"value": 4}, + } + } + }, + }, + ] + } + } + tweaks = {"node1": {"param3": 5}} + result = process_tweaks(graph_data, tweaks) + assert result == graph_data