def test_functions():
assert vp.sqrt(d_a).ccode("i") == "sqrt(a[i])"
assert vp.log(d_a).ccode("i") == "log(a[i])"
assert vp.cos(d_a).ccode("i") == "cos(a[i])"
assert vp.sin(d_a).ccode("i") == "sin(a[i])"
assert vp.exp(d_a).ccode("i") == "exp(a[i])"
assert vp.pow(d_a, 2).ccode("i") == "pow(a[i], 2)"
assert vp.function("test", d_a, 2).ccode("i") == "test(a[i], 2)"
def test_functions():
assert str(vp.function("test", va, 2)) == "test(a, 2)"
assert str(vp.sqrt(va)) == "sqrt(a)"
assert str(vp.sin(va)) == "sin(a)"
assert str(vp.cos(va)) == "cos(a)"
assert str(vp.exp(va)) == "exp(a)"
assert str(vp.log(va)) == "log(a)"
assert str(vp.abs(va)) == "fabs(a)"
assert str(vp.pow(va, 2)) == "pow(a, 2)"
assert str(vp.floor(va)) == "floor(a)"
def cos(expr: vecpy.base.Expr):
"""
Apply the `cos` function to a VecPy Expression.
Args:
expr: VecPy expression
Returns:
VecPy function
"""
return vecpy.function("cos", expr)
def floor(expr: vecpy.base.Expr):
"""
Apply the `floor` function to a VecPy Expression.
Args:
expr: VecPy expression
Returns:
VecPy function
"""
return vecpy.function("floor", expr)
def exp(expr: vecpy.base.Expr):
"""
Apply the `exp` function to a VecPy Expression.
Args:
expr: VecPy expression
Returns:
VecPy function
"""
return vecpy.function("exp", expr)
def sqrt(expr: vecpy.base.Expr):
"""
Apply the `sqrt` function to a VecPy Expression.
Args:
expr: VecPy expression
Returns:
VecPy function
"""
return vecpy.function("sqrt", expr)
def log(expr: vecpy.base.Expr):
"""
Apply the `log` function to a VecPy Expression.
Args:
expr: VecPy expression
Returns:
VecPy function
"""
return vecpy.function("log", expr)
def pow(expr: vecpy.base.Expr, exponent: vecpy.base.Expr):
"""
Apply the `pow` function to a VecPy Expression.
Args:
expr: VecPy expression
exponent: Exponent
Returns:
VecPy function
"""
return vecpy.function("pow", expr, exponent)