def test_get_constant_multiplied(self): factor = Factor(self.random_variables) factor.add_value([1, 3, 2], 10).add_value([2, 4, 2], 20).add_value([2, 5, 2], 30) factor = factor.get_constant_multiplied(2) self.assertEqual(factor.get_value([1, 3, 2]), 20) self.assertEqual(factor.get_value([2, 4, 2]), 40) self.assertEqual(factor.get_value([2, 5, 2]), 60) self.assertEqual(factor.get_value([1, 4, 2]), 0) self.assertEqual(factor.get_value([2, 3, 2]), 0)