starred/zen
1
0
Fork 0
mirror of https://github.com/gorules/zen.git synced 2026-10-05 16:02:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add batch evaluation for python and tests in workflows

Browse Source
This commit is contained in:
Stefan
2026-06-29 14:32:25 +02:00
parent 7805da79f7
commit 35f6519c71
6 changed files with 198 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bindings/python/test_async.py
+16
View File
@@ -79,6 +79,22 @@ class AsyncZenEngine(unittest.IsolatedAsyncioTestCase):
r = await functionDecision.async_evaluate({"input": 15})
self.assertEqual(r["result"]["output"], 30)
async def test_async_evaluate_batch(self):
engine = zen.ZenEngine({"loader": loader})
results = await engine.async_evaluate_batch([
("table.json", {"input": 12}),
("table.json", {"input": 2}),
("does-not-exist.json", {}),
])
self.assertEqual(len(results), 3)
self.assertTrue(results[0]["success"])
self.assertEqual(results[0]["data"]["result"]["output"], 10)
self.assertTrue(results[1]["success"])
self.assertEqual(results[1]["data"]["result"]["output"], 0)
self.assertFalse(results[2]["success"])
self.assertIsNotNone(results[2]["error"])
async def test_evaluate_graphs(self):
engine = zen.ZenEngine({"loader": graph_loader})
json_files = glob.glob("../../test-data/graphs/*.json")