def test_function_start(function):
"""
Assert function fixture includes decorations
"""
result = get_first_token(function.node).line
assert result == "@unittest.skip('not today')\n"
def test_with_decorators(first_node_with_tokens, lines, tokens):
result = Function(first_node_with_tokens, lines, tokens)
assert result.first_line_no == 4
assert get_first_token(
result.node
).line == "@pytest.mark.skip(reason='maths is too hard :D')\n"
assert get_last_token(result.node).line == ' assert result == 1\n'
assert len(result.lines) == 10
def test_act_marker_multi_line(function: Function) -> None:
"""
Act marker can be at end of bunch of lines
"""
result = function.load_act_node()
assert isinstance(result, ActNode)
assert result.block_type == ActNodeType.marked_act
assert get_first_token(result.node).line == ' validate_row(\n'
def test_strings(first_node_with_tokens, expected):
result = get_first_token(first_node_with_tokens)
assert isinstance(result, Token)
assert result.string == expected
def test(first_node_with_tokens):
result = get_first_token(first_node_with_tokens)
assert isinstance(result, Token)
assert result.string == 'def'