mirror of
https://github.com/tencentmusic/supersonic.git
synced 2025-12-10 19:51:00 +00:00
(improvement)(launcher)Add unit tests to the Text2SQLEval.
This commit is contained in:
@@ -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");
|
||||
|
||||
Reference in New Issue
Block a user