You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
13 lines
432 B
13 lines
432 B
from sympy.core.symbol import symbols
|
|
from sympy.codegen.pynodes import List
|
|
|
|
|
|
def test_List():
|
|
l = List(2, 3, 4)
|
|
assert l == List(2, 3, 4)
|
|
assert str(l) == "[2, 3, 4]"
|
|
x, y, z = symbols('x y z')
|
|
l = List(x**2,y**3,z**4)
|
|
# contrary to python's built-in list, we can call e.g. "replace" on List.
|
|
m = l.replace(lambda arg: arg.is_Pow and arg.exp>2, lambda p: p.base-p.exp)
|
|
assert m == [x**2, y-3, z-4]
|