def test_ok():
line_markers = LineMarkers(6, 7)
line_markers[0] = LineType.func_def
line_markers[1] = LineType.arrange
line_markers[2] = LineType.blank_line
line_markers[3] = LineType.act
line_markers[4] = LineType.blank_line
line_markers[5] = LineType._assert
result = line_markers.check_blank_lines()
assert isinstance(result, typing.Generator)
assert list(result) == []
def test_func_def():
"""
Function definition has some funky call args separated by a blank line
"""
line_markers = LineMarkers(3, 7)
line_markers[0] = LineType.func_def
line_markers[1] = LineType.blank_line
line_markers[2] = LineType.func_def
result = line_markers.check_blank_lines()
assert isinstance(result, typing.Generator)
assert list(result) == [
AAAError(
line_number=8,
offset=0,
text='AAA05 blank line in block',
),
]
def test_arrange():
line_markers = LineMarkers(8, 7)
line_markers[0] = LineType.func_def
line_markers[1] = LineType.arrange
line_markers[2] = LineType.blank_line
line_markers[3] = LineType.arrange
line_markers[4] = LineType.blank_line
line_markers[5] = LineType.act
line_markers[6] = LineType.blank_line
line_markers[7] = LineType._assert
result = line_markers.check_blank_lines()
assert isinstance(result, typing.Generator)
assert list(result) == [
AAAError(
line_number=9,
offset=0,
text='AAA05 blank line in block',
),
]
