def test_lines_to_statements(file_name): lines = add_context(file=file_name, lines=( 'line1\n', ' line2 \n', 'line3 part1\\\n', 'line3 part2\n', 'line \\ number 4\n', 'line5 part1\\\n', ' line5\\ part2\\\n', 'line5 part3\n', 'line6', )) expected = ( line_in_context('line1', context(file_name, 1)), line_in_context(' line2 ', context(file_name, 2)), line_in_context('line3 part1\nline3 part2', context(file_name, 3)), line_in_context('line \\ number 4', context(file_name, 5)), line_in_context( 'line5 part1\n line5\\ part2\nline5 part3', context(file_name, 6) ), line_in_context('line6', context(file_name, 9)), ) result = tuple(lines_to_statements(lines)) assert expected == result
def test_statement_to_object(file_name): statements = add_context(file_name, ( '# some comment', 'a_command with:args', )) expected_objects = ( comment(' some comment', context(file_name, 1)), command('a_command', 'with:args', context(file_name, 2)), ) for sttmt, expected in _it.izip(statements, expected_objects): result = statement_to_object(sttmt) assert expected == result
def test_add_context(file_name): lines = ( 'line1', 'line2', 'line3', ) expected = ( line_in_context(context=context(file=file_name, line=1), line='line1'), line_in_context(context=context(file=file_name, line=2), line='line2'), line_in_context(context=context(file=file_name, line=3), line='line3'), ) result = tuple(add_context(file=file_name, lines=lines)) assert expected == result
def test_add_context_nested(file_name, some_context): lines = ( 'line1', 'line2', 'line3', ) ctx = ( nested_context(file_name, cnt, some_context) for cnt in _it.count(1) ) expected = ( line_in_context(context=next(ctx), line='line1'), line_in_context(context=next(ctx), line='line2'), line_in_context(context=next(ctx), line='line3'), ) result = tuple( add_context(file=file_name, lines=lines, nest_in=some_context) ) assert expected == result