def test_add_merge(self): dai = DialogueActItem(dai='inform(food=chinese)') dacn = DialogueActConfusionNetwork() dacn.add_merge(0.5, dai, combine='add') self.assertEqual(dacn._get_prob([0]), 0.5) dacn.add_merge(0.5, dai, combine='add') self.assertEqual(dacn._get_prob([0]), 1.0)
def test_add_merge(self): dai = DialogueActItem(dai='inform(food=chinese)') dacn = DialogueActConfusionNetwork() dacn.add_merge(0.5, dai, combine='add') self.assertEqual(dacn._get_prob([0]), 0.5) dacn.add_merge(0.5, dai, combine='add') self.assertEqual(dacn._get_prob([0]), 1.0)
def test_get_prob(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)')) self.assertAlmostEqual(dacn._get_prob([0, 1, 1]), 0.2 * 0.3 * 0.9) self.assertAlmostEqual(dacn._get_prob([0, 0, 0]), 0.2 * 0.7 * 0.1)
def test_get_prob(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)')) self.assertAlmostEqual(dacn._get_prob([0, 1, 1]), 0.2 * 0.3 * 0.9) self.assertAlmostEqual(dacn._get_prob([0, 0, 0]), 0.2 * 0.7 * 0.1)