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)'))