Don't print both start and end position for zero-length ranges.

whitequark · whitequark · commit bb6b14375686 · 2015-07-22T00:04:26.000+03:00
diff --git a/pythonparser/source.py b/pythonparser/source.py
@@ -159,9 +159,13 @@ def __str__(self):
         """
         Returns a Clang-style string representation of the beginning of this range.
         """
-        return "%s:%d:%d-%d:%d" % (self.source_buffer.name,
-                                self.line(), self.column() + 1,
-                                self.end().line(), self.end().column() + 1)
+        if self.begin_pos != self.end_pos:
+            return "%s:%d:%d-%d:%d" % (self.source_buffer.name,
+                                    self.line(), self.column() + 1,
+                                    self.end().line(), self.end().column() + 1)
+        else:
+            return "%s:%d:%d" % (self.source_buffer.name,
+                                 self.line(), self.column() + 1)
 
     def __eq__(self, other):
         """
diff --git a/pythonparser/test/test_source.py b/pythonparser/test/test_source.py
@@ -90,7 +90,8 @@ def test_source_lines(self):
         self.assertEqual(["line two\n", "\n"], self.range(9, 18).source_lines())
 
     def test___str__(self):
-        self.assertEqual("<input>:2:1-2:1", str(self.range(9, 9)))
+        self.assertEqual("<input>:2:1-2:2", str(self.range(9, 10)))
+        self.assertEqual("<input>:2:1", str(self.range(9, 9)))
 
     def test___ne__(self):
         self.assertTrue(self.range(0,0) != self.range(0,1))
