def test_get_sum_marginalized(self):
init_factor = Factor(self.random_variables)
init_factor.add_value([1, 3, 2], 10).add_value([1, 4, 2],
5).add_value([1, 5, 2],
2)
init_factor.add_value([2, 4, 2], 20).add_value([2, 5, 2], 30)
factor = init_factor.get_sum_marginalized([self.random_variables[0]])
self.assertEqual(25, factor.get_value({'y': 4, 'z': 2}))
self.assertEqual(32, factor.get_value({'y': 5, 'z': 2}))
self.assertEqual(10, factor.get_value({'y': 3, 'z': 2}))
factor = init_factor.get_sum_marginalized(self.random_variables)
self.assertEqual(67, factor.get_value([]))
