浏览代码

Update unit_test: add more test for test_response_type_detector

StrayWarrior 1 天之前
父节点
当前提交
83c9a99edb
共有 1 个文件被更改,包括 5 次插入0 次删除
  1. 5 0
      unit_test.py

+ 5 - 0
unit_test.py

@@ -217,5 +217,10 @@ def test_initiative_conversation(test_env):
 def test_response_type_detector(test_env):
 def test_response_type_detector(test_env):
     case1 = '大哥,那可得提前了解下天气,以便安排行程~我帮您查查明天北京天气?'
     case1 = '大哥,那可得提前了解下天气,以便安排行程~我帮您查查明天北京天气?'
     assert ResponseTypeDetector.is_chinese_only(case1) == True
     assert ResponseTypeDetector.is_chinese_only(case1) == True
+    assert ResponseTypeDetector.if_message_suitable_for_voice(case1) == False
     case2 = 'hi'
     case2 = 'hi'
     assert ResponseTypeDetector.is_chinese_only(case2) == False
     assert ResponseTypeDetector.is_chinese_only(case2) == False
+    case3 = '这是链接:http://domain.com'
+    assert ResponseTypeDetector.is_chinese_only(case3) == False
+    case4 = '大哥,那可得提前了解下天气'
+    assert ResponseTypeDetector.if_message_suitable_for_voice(case4) == True