def test_vectorize():
vsin1 = vectorize(Integer(0))(sin)
pytest.raises(TypeError, lambda: vectorize(Rational(1, 2))(sin))
@vectorize()
def vsin2(x):
return sin(x)
@vectorize(0, "evaluate")
def vsin3(x, evaluate=True):
return sin(x, evaluate=evaluate)
@vectorize(0, "evaluate")
def vlog1(x, base=None, evaluate=True):
return log(x, base, evaluate=evaluate)
@vectorize(0, 3, "boo")
def vlog2(x, base=S.Exp1, **kwargs):
return log(x, base)
@vectorize(0, 1)
def vdiff(f, y):
return diff(f, y)
f, g = map(Function, 'fg')
assert vsin1([1, x, y]) == [sin(1), sin(x), sin(y)]
assert vsin1(tuple([1, tuple([x, y])])) == [sin(1), [sin(x), sin(y)]]
assert vsin1((1, x, y)) == [sin(1), sin(x), sin(y)]
assert vsin2([1, x, y]) == [sin(1), sin(x), sin(y)]
assert (vsin3([pi, 2 * pi, pi / 2],
evaluate=[True, False]) == [[0, sin(pi, evaluate=False)],
[0,
sin(2 * pi, evaluate=False)],
[1,
sin(pi / 2, evaluate=False)]])
assert (vlog1([1, 2], evaluate=[True, False],
base=2) == [[0, log(1, 2, evaluate=False)],
[1, log(2, 2, evaluate=False)]])
assert vlog2([1, 2]) == [0, log(2)]
assert (vdiff([f(x, y, z), g(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)
]])
def test_core_multidimensional():
for c in (vectorize, vectorize(0)):
check(c)
def test_core_multidimensional():
for c in (vectorize, vectorize(0)):
check(c)