def test_random_symbols(): X, Y = Normal('X', 0, 1), Normal('Y', 0, 1) assert set(random_symbols(2*X + 1)) == set((X,)) assert set(random_symbols(2*X + Y)) == set((X, Y)) assert set(random_symbols(2*X + Y.symbol)) == set((X,)) assert set(random_symbols(2)) == set()
def test_random_symbols(): X, Y = Normal('X', 0, 1), Normal('Y', 0, 1) assert set(random_symbols(2*X + 1)) == set((X,)) assert set(random_symbols(2*X + Y)) == set((X, Y)) assert set(random_symbols(2*X + Y.symbol)) == set((X,)) assert set(random_symbols(2)) == set()
def test_joint_normal_separate(): rvs = JointNormalSeparate(['ETA(1)', 'ETA(2)'], [0, 0], [[1, 0], [0, 1]]) assert rvs[0].name == 'ETA(1)' assert rvs[1].name == 'ETA(2)' assert rvs[0].pspace.distribution.mu == sympy.Matrix([[0], [0]]) assert rvs[0].pspace.distribution.sigma == sympy.Matrix([[1, 0], [0, 1]]) assert stats.random_symbols(rvs[0]) == [rvs[0]] assert stats.random_symbols(rvs[1]) == [rvs[1]] rvs2 = JointNormalSeparate(['ETA(3)', 'ETA(4)'], [1, 1], [[6, 3], [4, 5]]) # Check that previously created rvs are still intact assert rvs[0].name == 'ETA(1)' assert rvs[1].name == 'ETA(2)' assert rvs[0].pspace.distribution.mu == sympy.Matrix([[0], [0]]) assert rvs[0].pspace.distribution.sigma == sympy.Matrix([[1, 0], [0, 1]]) assert stats.random_symbols(rvs[0]) == [rvs[0]] assert stats.random_symbols(rvs[1]) == [rvs[1]] assert len(rvs2) == 2