def test_number_invalid(self, parser, ctx):
"""Test that an unknown number correctly raises and exception"""
number = blackbirdParser.NumberContext(parser, ctx)
number.getText = lambda: ""
with pytest.raises(ValueError, match="Unknown number"):
_number(number)
def test_number_pi(self, parser, ctx):
"""Test that a Blackbird pi literal is properly converted to a Python type"""
number = blackbirdParser.NumberContext(parser, ctx)
number.PI = lambda: True
assert _number(number) == np.pi
def test_number_float(self, parser, ctx, n, expected):
"""Test that a Blackbird float is properly converted to a Python type"""
number = blackbirdParser.NumberContext(parser, ctx)
number.FLOAT = lambda: True
number.getText = lambda: n
assert _number(number) == expected
def test_number_complex(self, parser, ctx, n, expected):
"""Test that a Blackbird complex is properly converted to a Python type"""
number = blackbirdParser.NumberContext(parser, ctx)
number.COMPLEX = lambda: True
number.getText = lambda: n
assert _number(number) == expected
def test_number_int(self, parser, ctx, n, expected):
"""Test that a Blackbird int is properly converted to a Python type"""
number = blackbirdParser.NumberContext(parser, ctx)
number.INT = lambda: True
number.getText = lambda: n
assert np.isclose(_number(number), expected)
