def test_equations():
A = System()
A.x = 2*A.y
assert A[...].equations[A.x] == [sympy.Eq(A.x, 2*A.y)]
assert A[A.x] == [sympy.Eq(A.x, 2*A.y)]
A[A.x] = 20*A.y
assert A[A.x] == [sympy.Eq(A.x, 20*A.y)]
A.x = sympy.Eq(2*A.y, 3*A.z)
assert A[A.x] == [sympy.Eq(A.x, 2*A.y), sympy.Eq(A.x, 3*A.z)]
A.x = (4*A.x1, 5*A.x2, 6*A.x3)
assert A[A.x] == [sympy.Eq(A.x, 4*A.x1), sympy.Eq(A.x, 5*A.x2),
sympy.Eq(A.x, 6*A.x3)]
A[1] = 2*A.x
assert A[...].equations[A[1]] == [sympy.Eq(A[1], 2*A.x)]
assert A[A[1]] == [sympy.Eq(A[1], 2*A.x)]
A[A[1]] = 20*A.x
assert A[A[1]] == [sympy.Eq(A[1], 20*A.x)]
def test_userdict():
A = System()
A['x'] = 5
A.x = 4
assert A['x'] == 5
assert isinstance(A[A.x][0], sympy.Equality)
A['1'] = 6
A[1] = 7
assert A['1'] == 6
assert isinstance(A[A[1]][0], sympy.Equality)
def test_userdict():
A = System()
A['x'] = 5
A.x = 4
assert A['x'] == 5
assert isinstance(A[A.x][0], sympy.Equality)
A['1'] = 6
A[1] = 7
assert A['1'] == 6
assert isinstance(A[A[1]][0], sympy.Equality)
def test_equations():
A = System()
A.x = 2 * A.y
assert A[...].equations[A.x] == [sympy.Eq(A.x, 2 * A.y)]
assert A[A.x] == [sympy.Eq(A.x, 2 * A.y)]
A[A.x] = 20 * A.y
assert A[A.x] == [sympy.Eq(A.x, 20 * A.y)]
A.x = sympy.Eq(2 * A.y, 3 * A.z)
assert A[A.x] == [sympy.Eq(A.x, 2 * A.y), sympy.Eq(A.x, 3 * A.z)]
A.x = (4 * A.x1, 5 * A.x2, 6 * A.x3)
assert A[A.x] == [
sympy.Eq(A.x, 4 * A.x1),
sympy.Eq(A.x, 5 * A.x2),
sympy.Eq(A.x, 6 * A.x3)
]
A[1] = 2 * A.x
assert A[...].equations[A[1]] == [sympy.Eq(A[1], 2 * A.x)]
assert A[A[1]] == [sympy.Eq(A[1], 2 * A.x)]
A[A[1]] = 20 * A.x
assert A[A[1]] == [sympy.Eq(A[1], 20 * A.x)]
def test_contains():
A = System()
A.x = 2*A.y
A['z'] = 'zee'
A[1] = 3*A[2]
A['3'] = 'three'
assert A.x in A
assert A.y not in A
assert 'z' in A
assert 'zee' not in A
assert A[1] in A
assert A[2] not in A
assert '3' in A
assert 'three' not in A
def test_contains():
A = System()
A.x = 2 * A.y
A['z'] = 'zee'
A[1] = 3 * A[2]
A['3'] = 'three'
assert A.x in A
assert A.y not in A
assert 'z' in A
assert 'zee' not in A
assert A[1] in A
assert A[2] not in A
assert '3' in A
assert 'three' not in A