Use the same function to detect newlines in the diff parser.

This commit is contained in:
Dave Halter
2017-01-23 09:56:38 +01:00
parent 1caa2ceafa
commit 01b25efea1
2 changed files with 2 additions and 2 deletions

View File

@@ -228,7 +228,7 @@ class DiffParser(object):
def _get_old_line_stmt(self, old_line): def _get_old_line_stmt(self, old_line):
leaf = self._module.get_leaf_for_position((old_line, 0), include_prefixes=True) leaf = self._module.get_leaf_for_position((old_line, 0), include_prefixes=True)
if leaf.type == 'newline': if _ends_with_newline(leaf):
leaf = leaf.get_next_leaf() leaf = leaf.get_next_leaf()
if leaf.get_start_pos_of_prefix()[0] == old_line: if leaf.get_start_pos_of_prefix()[0] == old_line:
node = leaf node = leaf

View File

@@ -265,7 +265,7 @@ def test_issues_with_error_leaves(differ):
str str
''') ''')
differ.initialize(code) differ.initialize(code)
differ.parse(code2, parsers=2, copies=0, expect_error_leaves=True) differ.parse(code2, parsers=1, copies=1, expect_error_leaves=True)
def test_unfinished_nodes(differ): def test_unfinished_nodes(differ):