From ac6b28ebb73efb773bb6a242b15d008cd4f60212 Mon Sep 17 00:00:00 2001 From: jerryjzhang Date: Fri, 2 Aug 2024 18:08:54 +0800 Subject: [PATCH] (improvement)(launcher)Add unit tests to the `Text2SQLEval`. --- .../supersonic/evaluation/Text2SQLEval.java | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/launchers/standalone/src/test/java/com/tencent/supersonic/evaluation/Text2SQLEval.java b/launchers/standalone/src/test/java/com/tencent/supersonic/evaluation/Text2SQLEval.java index eaf7f2e4e..ac91c032b 100644 --- a/launchers/standalone/src/test/java/com/tencent/supersonic/evaluation/Text2SQLEval.java +++ b/launchers/standalone/src/test/java/com/tencent/supersonic/evaluation/Text2SQLEval.java @@ -79,6 +79,24 @@ public class Text2SQLEval extends BaseTest { assert result.getQueryResults().size() == 2; } + @Test + public void test_filter_compare() throws Exception { + QueryResult result = submitNewChat("alice和lucy过去半个月哪一位的总停留时长更高", agentId); + assert result.getQueryColumns().size() == 2; + assert result.getQueryColumns().get(0).getName().contains("用户"); + assert result.getQueryColumns().get(1).getName().contains("停留时长"); + assert result.getQueryResults().size() >= 1; + } + + @Test + public void test_term() throws Exception { + QueryResult result = submitNewChat("过去半个月核心用户的总停留时长", agentId); + assert result.getQueryColumns().size() == 2; + assert result.getQueryColumns().get(0).getName().contains("用户"); + assert result.getQueryColumns().get(1).getName().contains("停留时长"); + assert result.getQueryResults().size() == 2; + } + public static Agent getLLMAgent(boolean enableMultiturn) { Agent agent = new Agent(); agent.setName("Agent for Test");