mirror of
https://github.com/PyCQA/flake8.git
synced 2026-04-02 03:46:53 +00:00
Add a few more unit tests around FileProcessor
This commit is contained in:
parent
7de8a058dd
commit
4ea161ff9c
1 changed files with 42 additions and 0 deletions
|
|
@ -167,3 +167,45 @@ def test_build_ast():
|
|||
|
||||
module = file_processor.build_ast()
|
||||
assert isinstance(module, ast.Module)
|
||||
|
||||
|
||||
def test_next_logical_line_updates_the_previous_logical_line():
|
||||
"""Verify that we update our tracking of the previous logical line."""
|
||||
file_processor = processor.FileProcessor('-', options_from(), lines=[
|
||||
'a = 1\n'
|
||||
])
|
||||
|
||||
file_processor.indent_level = 1
|
||||
file_processor.logical_line = 'a = 1'
|
||||
assert file_processor.previous_logical == ''
|
||||
assert file_processor.previous_indent_level is 0
|
||||
|
||||
file_processor.next_logical_line()
|
||||
assert file_processor.previous_logical == 'a = 1'
|
||||
assert file_processor.previous_indent_level == 1
|
||||
|
||||
|
||||
def test_visited_new_blank_line():
|
||||
"""Verify we update the number of blank lines seen."""
|
||||
file_processor = processor.FileProcessor('-', options_from(), lines=[
|
||||
'a = 1\n'
|
||||
])
|
||||
|
||||
assert file_processor.blank_lines == 0
|
||||
file_processor.visited_new_blank_line()
|
||||
assert file_processor.blank_lines == 1
|
||||
|
||||
|
||||
def test_inside_multiline():
|
||||
"""Verify we update the line number and reset multiline."""
|
||||
file_processor = processor.FileProcessor('-', options_from(), lines=[
|
||||
'a = 1\n'
|
||||
])
|
||||
|
||||
assert file_processor.multiline is False
|
||||
assert file_processor.line_number == 0
|
||||
with file_processor.inside_multiline(10):
|
||||
assert file_processor.multiline is True
|
||||
assert file_processor.line_number == 10
|
||||
|
||||
assert file_processor.multiline is False
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue