(improvement)(launcher)Add unit tests to the Text2SQLEval.

This commit is contained in:
jerryjzhang
2024-08-02 18:08:54 +08:00
parent 53a9f7c451
commit ac6b28ebb7

View File

@@ -79,6 +79,24 @@ public class Text2SQLEval extends BaseTest {
assert result.getQueryResults().size() == 2; 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) { public static Agent getLLMAgent(boolean enableMultiturn) {
Agent agent = new Agent(); Agent agent = new Agent();
agent.setName("Agent for Test"); agent.setName("Agent for Test");