source.Range.column_range: clamp to first line.

m-labs · Sep 2, 2015 · 3f3ae13 · 3f3ae13
1 parent d2e0d1b
commit 3f3ae13
Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/pythonparser/source.py b/pythonparser/source.py
@@ -125,9 +125,13 @@ def column(self):
     def column_range(self):
         """
         Returns a [*begin*, *end*) tuple describing the range of columns spanned
-        by this range.
+        by this range. If range spans more than one line, returned *end* is
+        the last column of the line.
         """
-        return self.begin().column(), self.end().column()
+        if self.begin().line() == self.end().line():
+            return self.begin().column(), self.end().column()
+        else:
+            return self.begin().column(), len(self.begin().source_line()) - 1
 
     def line(self):
         """

diff --git a/pythonparser/test/test_source.py b/pythonparser/test/test_source.py
@@ -61,6 +61,7 @@ def test_column_range(self):
         self.assertEqual((2,2), self.range(2, 2).column_range())
         self.assertEqual((0,2), self.range(9, 11).column_range())
         self.assertEqual((2,2), self.range(11, 11).column_range())
+        self.assertEqual((0,8), self.range(0, 11).column_range())
 
     def test_line(self):
         self.assertEqual(1, self.range(2, 2).line())