Skip to content

Commit e64a7f5

Browse files
committed
adjust line_info tests to include sources
1 parent 651328f commit e64a7f5

File tree

1 file changed

+18
-6
lines changed

1 file changed

+18
-6
lines changed

tests/test_parsy.py

Lines changed: 18 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -606,6 +606,18 @@ def foo():
606606
],
607607
)
608608

609+
source = 'aaaaa'
610+
self.assertEqual(
611+
foo.many().parse(Stream("AB\nCD", source=source)),
612+
[
613+
("A", (source, 0, 0)),
614+
("B", (source, 0, 1)),
615+
("\n", (source, 0, 2)),
616+
("C", (source, 1, 0)),
617+
("D", (source, 1, 1)),
618+
],
619+
)
620+
609621
def test_should_fail(self):
610622
not_a_digit = digit.should_fail("not a digit") >> regex(r".*")
611623

@@ -700,12 +712,12 @@ def foo():
700712

701713
class TestUtils(unittest.TestCase):
702714
def test_line_info_at(self):
703-
text = Stream("abc\ndef")
704-
self.assertEqual(line_info_at(text, 0), (None, 0, 0))
705-
self.assertEqual(line_info_at(text, 2), (None, 0, 2))
706-
self.assertEqual(line_info_at(text, 3), (None, 0, 3))
707-
self.assertEqual(line_info_at(text, 4), (None, 1, 0))
708-
self.assertEqual(line_info_at(text, 7), (None, 1, 3))
715+
text = Stream("abc\ndef", source='aaaa')
716+
self.assertEqual(line_info_at(text, 0), ('aaaa', 0, 0))
717+
self.assertEqual(line_info_at(text, 2), ('aaaa', 0, 2))
718+
self.assertEqual(line_info_at(text, 3), ('aaaa', 0, 3))
719+
self.assertEqual(line_info_at(text, 4), ('aaaa', 1, 0))
720+
self.assertEqual(line_info_at(text, 7), ('aaaa', 1, 3))
709721
self.assertRaises(ValueError, lambda: line_info_at(text, 8))
710722

711723

0 commit comments

Comments
 (0)