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
