def test_computeMarginal1 (self): # Place into an array factor_array = [] factor_array.append(self.factorA) factor_array.append(self.factorB) factor_array.append(self.factorC) expectedFactor = pgmf.Factor(np.array([2, 3]), np.array([2, 2]), np.array([0.0858, 0.0468, 0.1342, 0.7332])) resultFactor = pgmf.computeMarginal(np.array([2,3]), factor_array, np.array([1,2])) np.testing.assert_array_equal(resultFactor.varbs, expectedFactor.varbs) np.testing.assert_array_equal(resultFactor.card, expectedFactor.card) np.testing.assert_array_almost_equal(resultFactor.vals, expectedFactor.vals, decimal = 4)