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_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