Exemple #1
0
    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)