def test_lines_counts(self): lines = [ 'line1\n', 'line2\n', 'line3\n', '\n', '\n', 'line4\n', ] py_node = FileNode('test.py', ''.join(lines)) assert (len(lines), len(lines)) == py_node.lines() assert (len(lines), len(lines) - 2) == py_node.lines(count_empty=True)
def test_lines_no_newline(self): py_node = FileNode('test.py', 'oneline') assert (1, 1) == py_node.lines() assert (1, 1) == py_node.lines(count_empty=True)