Skip to content

Commit 6f54191

Browse files
fix: 修复python测试 (#590)
1 parent 75eb9aa commit 6f54191

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

‎data-agent-management/src/test/java/com/alibaba/cloud/ai/dataagent/workflow/node/python/PythonExecuteNodeTest.java‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -211,9 +211,9 @@ void apply_withMultipleSqlResults_passesOrderedResultSetsToPython() throws Excep
211211
assertTrue(result.containsKey(PYTHON_EXECUTE_NODE_OUTPUT));
212212
assertNotNull(result.get(PYTHON_EXECUTE_NODE_OUTPUT));
213213

214-
ArgumentCaptor<CodePoolExecutorService.TaskRequest> request = ArgumentCaptor
215-
.forClass(CodePoolExecutorService.TaskRequest.class);
216-
verify(codePoolExecutor).runTask(request.capture());
214+
ArgumentCaptor<PythonCodeExecutorService.TaskRequest> request = ArgumentCaptor
215+
.forClass(PythonCodeExecutorService.TaskRequest.class);
216+
verify(pythonCodeExecutor).runTask(request.capture());
217217
String input = request.getValue().input();
218218
String message = "Expected ordered sales and engineering result sets, but Python received: " + input;
219219
assertTrue(input.startsWith("[[{"), message);

0 commit comments

Comments
 (0)