| 123456789101112131415161718192021222324 |
- from sympy.core.function import (Derivative, Function, diff)
- from sympy.core.symbol import symbols
- from sympy.functions.elementary.trigonometric import sin
- from sympy.core.multidimensional import vectorize
- x, y, z = symbols('x y z')
- f, g, h = list(map(Function, 'fgh'))
- def test_vectorize():
- @vectorize(0)
- def vsin(x):
- return sin(x)
- assert vsin([1, x, y]) == [sin(1), sin(x), sin(y)]
- @vectorize(0, 1)
- def vdiff(f, y):
- return diff(f, y)
- assert vdiff([f(x, y, z), g(x, y, z), h(x, y, z)], [x, y, z]) == \
- [[Derivative(f(x, y, z), x), Derivative(f(x, y, z), y),
- Derivative(f(x, y, z), z)], [Derivative(g(x, y, z), x),
- Derivative(g(x, y, z), y), Derivative(g(x, y, z), z)],
- [Derivative(h(x, y, z), x), Derivative(h(x, y, z), y), Derivative(h(x, y, z), z)]]
|
