예제 #1
0
파일: test_da.py 프로젝트: UFAL-DSG/alex
    def test_get_best_da(self):
        dacn = DialogueActConfusionNetwork()
        dacn.add(0.2, DialogueActItem(dai='inform(food=chinese)'))
        dacn.add(0.7, DialogueActItem(dai='inform(food=czech)'))
        dacn.add(0.1, DialogueActItem(dai='inform(food=russian)'))

        da = dacn.get_best_da()
        self.assertEqual(len(da), 1)
        self.assertEqual(da.dais[0], DialogueActItem(dai='inform(food=czech)'))

        dacn = DialogueActConfusionNetwork()
        dacn.add(0.2, DialogueActItem(dai='inform(food=chinese)'))
        dacn.add(0.3, DialogueActItem(dai='inform(food=czech)'))
        dacn.add(0.7, DialogueActItem(dai='inform(food=null)'))
        dacn.add(0.1, DialogueActItem(dai='inform(food=russian)'))

        da = dacn.get_best_nonnull_da()
        self.assertEqual(len(da), 1)
        self.assertEqual(da.dais[0], DialogueActItem(dai='inform(food=null)'))
예제 #2
0
파일: test_da.py 프로젝트: henrypig/alex-1
    def test_get_best_da(self):
        dacn = DialogueActConfusionNetwork()
        dacn.add(0.2, DialogueActItem(dai='inform(food=chinese)'))
        dacn.add(0.7, DialogueActItem(dai='inform(food=czech)'))
        dacn.add(0.1, DialogueActItem(dai='inform(food=russian)'))

        da = dacn.get_best_da()
        self.assertEqual(len(da), 1)
        self.assertEqual(da.dais[0], DialogueActItem(dai='inform(food=czech)'))

        dacn = DialogueActConfusionNetwork()
        dacn.add(0.2, DialogueActItem(dai='inform(food=chinese)'))
        dacn.add(0.3, DialogueActItem(dai='inform(food=czech)'))
        dacn.add(0.7, DialogueActItem(dai='inform(food=null)'))
        dacn.add(0.1, DialogueActItem(dai='inform(food=russian)'))

        da = dacn.get_best_nonnull_da()
        self.assertEqual(len(da), 1)
        self.assertEqual(da.dais[0], DialogueActItem(dai='inform(food=null)'))