def test_none():
line_markers = LineMarkers(2, 0)
line_markers[0] = LineType.func_def
line_markers[1] = LineType.act # do_thing()
result = line_markers.check_act_assert_spacing()
assert isinstance(result, Generator)
assert list(result) == []
def test_comment_before_assert():
"""
Comment before Assert passes
"""
line_markers = LineMarkers(8, 5)
line_markers[0] = LineType.func_def
line_markers[1] = LineType.arrange # x = 1
line_markers[2] = LineType.arrange # y = 2
line_markers[3] = LineType.blank_line
line_markers[4] = LineType.act # result = x + y
line_markers[5] = LineType.blank_line
line_markers[6] = LineType.unprocessed # Sum x and y
line_markers[7] = LineType._assert # assert result == 2
result = line_markers.check_act_assert_spacing()
assert isinstance(result, Generator)
assert list(result) == []
def test_no_gap():
"""
No gap raises - error points at act block because no spaces
"""
line_markers = LineMarkers(6, 5)
line_markers[0] = LineType.func_def
line_markers[1] = LineType.arrange # x = 1
line_markers[2] = LineType.blank_line
line_markers[3] = LineType.act # result = x + 3
line_markers[4] = LineType.unprocessed # Sum do stuff
line_markers[5] = LineType._assert # assert result == 4
result = line_markers.check_act_assert_spacing()
assert isinstance(result, Generator)
assert list(result) == [
AAAError(
line_number=9,
offset=0,
text='AAA04 expected 1 blank line before Assert block, found none',
),
]
def test_too_big_gap():
"""
Multiple BL raises. First extra line is pointed to
"""
line_markers = LineMarkers(8, 5)
line_markers[0] = LineType.func_def
line_markers[1] = LineType.arrange # x = 1
line_markers[2] = LineType.blank_line
line_markers[3] = LineType.unprocessed # Sum do stuff
line_markers[4] = LineType.act # result = x + 3
line_markers[5] = LineType.blank_line
line_markers[6] = LineType.blank_line
line_markers[7] = LineType._assert # assert result == 4
result = line_markers.check_act_assert_spacing()
assert isinstance(result, Generator)
assert list(result) == [
AAAError(
line_number=11,
offset=0,
text='AAA04 expected 1 blank line before Assert block, found 2',
),
]
