add .span() test

Author: tsani

tests/test_parsy.py

@@ -8,6 +8,7 @@
 from parsy import (
     ParseError,
     Stream,
+    SourceSpan,
     alt,
     any_char,
     char_from,
@@ -209,6 +210,21 @@ def test_mark(self):
         self.assertEqual(letters, ["q", "w", "e", "r"])
         self.assertEqual(end, (1, 4))
 
+    def test_span(self):
+        parser = (letter.many().span() << string("\n")).many()
+        source='sample'
+
+        lines = parser.parse(Stream("asdf\nqwer\n", source=source))
+
+        self.assertEqual(len(lines), 2)
+
+        (span, letters) = lines[0]
+        self.assertEqual(span, SourceSpan(source, (0, 0), (0, 4)))
+        self.assertEqual(letters, ['a', 's', 'd', 'f'])
+
+        (span, letters) = lines[1]
+        self.assertEqual(span, SourceSpan(source, (1, 0), (1, 4)))
+
     def test_tag(self):
         parser = letter.many().concat().tag("word")
         self.assertEqual(