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)