def test_newlines():
for toplevel_def in ('def bar():', 'class Bar(object):'):
for num_newlines in (0, 1, 3, 4):
newlines = Newlines(
PythonFile.from_statement(TOPLEVEL %
('\n' * num_newlines, toplevel_def)))
nits = list(newlines.nits())
assert len(nits) == 1
assert nits[0].code == 'T302'
assert nits[0].severity == Nit.ERROR
newlines = Newlines(
PythonFile.from_statement(TOPLEVEL % ('\n\n', toplevel_def)))
assert len(list(newlines.nits())) == 0
def test_classdefs():
newlines = Newlines(PythonFile.from_statement(GOOD_CLASS_DEF_1))
assert len(list(newlines.nits())) == 0
newlines = Newlines(PythonFile.from_statement(GOOD_CLASS_DEF_2))
assert len(list(newlines.nits())) == 0
newlines = Newlines(PythonFile.from_statement(BAD_CLASS_DEF_1))
nits = list(newlines.nits())
assert len(nits) == 1
assert nits[0].code == 'T301'
assert nits[0]._line_number == 4
assert nits[0].severity == Nit.ERROR
newlines = Newlines(PythonFile.from_statement(BAD_CLASS_DEF_2))
nits = list(newlines.nits())
assert len(nits) == 1
assert nits[0].code == 'T301'
assert nits[0]._line_number == 7
assert nits[0].severity == Nit.ERROR
