Fix some code_lines issues

This commit is contained in:
Dave Halter
2018-03-17 19:41:26 +01:00
parent 094affaf84
commit 60da6034c0
7 changed files with 17 additions and 15 deletions

View File

@@ -165,7 +165,7 @@ def test_get_line_code(Script):
# On custom code
first_line = 'def foo():\n'
line = ' foo'
code = '%s%s' % (first_line, line)
code = first_line + line
assert get_line_code(code) == first_line
# With before/after

View File

@@ -84,5 +84,5 @@ def test_namedtuple_goto_definitions(Script):
d1, = Script(source).goto_definitions()
assert d1.get_line_code() == "class Foo(tuple):"
assert d1.get_line_code() == "class Foo(tuple):\n"
assert d1.module_path is None