def test_function_invalid(self, parser, ctx):
"""Test that an invalid Blackbird function raises the correct exception"""
func = blackbirdParser.FunctionContext(parser, ctx)
func.getText = lambda: "Hello"
with pytest.raises(NameError, match="Unknown function Hello"):
_func(func, None)
def test_function_arctanh(self, parser, ctx, num):
"""Test that a Blackbird arctanh function is properly called"""
n = "0.543"
expected = float(n)
func = blackbirdParser.FunctionContext(parser, ctx)
func.ARCTANH = lambda: True
expression = num(n, num_type='float')
assert _func(func, expression) == np.arctanh(expected)
def test_function_arccos(self, parser, ctx, num):
"""Test that a Blackbird arccos function is properly called"""
n = "0.543"
expected = float(n)
func = blackbirdParser.FunctionContext(parser, ctx)
func.ARCCOS = lambda: True
expression = num(n)
assert np.allclose(_func(func, expression), np.arccos(expected))
def test_function_sqrt(self, parser, ctx, n, expected, num):
"""Test that a Blackbird sqrt function is properly called"""
func = blackbirdParser.FunctionContext(parser, ctx)
func.SQRT = lambda: True
expression = num(n)
assert _func(func, expression) == np.sqrt(expected)
def test_function_arccosh(self, parser, ctx, n, expected, num):
"""Test that a Blackbird arccosh function is properly called"""
func = blackbirdParser.FunctionContext(parser, ctx)
func.ARCCOSH = lambda: True
expression = num(n)
assert _func(func, expression) == np.arccosh(expected)
def test_function_tan(self, parser, ctx, n, expected, num):
"""Test that a Blackbird tan function is properly called"""
func = blackbirdParser.FunctionContext(parser, ctx)
func.TAN = lambda: True
expression = num(n, num_type='float')
assert _func(func, expression) == np.tan(expected)
def test_function_arcsinh(self, parser, ctx, n, expected, num):
"""Test that a Blackbird arcsinh function is properly called"""
func = blackbirdParser.FunctionContext(parser, ctx)
func.ARCSINH = lambda: True
expression = num(n)
assert np.allclose(_func(func, expression), np.arcsinh(expected))
def test_function_cosh(self, parser, ctx, n, expected, num):
"""Test that a Blackbird cosh function is properly called"""
func = blackbirdParser.FunctionContext(parser, ctx)
func.COSH = lambda: True
expression = num(n)
assert np.isclose(_func(func, expression), np.cosh(expected))
