diff options
Diffstat (limited to 'test/test_text.py')
| -rw-r--r-- | test/test_text.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/test/test_text.py b/test/test_text.py index 0ac7767..2c0be3b 100644 --- a/test/test_text.py +++ b/test/test_text.py @@ -203,6 +203,24 @@ class TestText(unittest.TestCase): self.assertEqual(f(txt , value, ">") , (None, 0)) self.assertEqual(f(txt , "<" , value), (None, 0)) + def test_extr(self, f=text.extr): + txt = "<a><b>" + self.assertEqual(f(txt, "X", ">"), "") + self.assertEqual(f(txt, "<", "X"), "") + self.assertEqual(f(txt, "<", ">"), "a") + self.assertEqual(f(txt, "><", ">"), "b") + + # 'default' argument + self.assertEqual(f(txt, "<", "X", None), None) + self.assertEqual(f(txt, "<", "X", default=None), None) + self.assertEqual(f(txt, "<", "X", default=()), ()) + + # invalid arguments + for value in INVALID: + self.assertEqual(f(value, "<" , ">") , "") + self.assertEqual(f(txt , value, ">") , "") + self.assertEqual(f(txt , "<" , value), "") + def test_rextract(self, f=text.rextract): txt = "<a><b>" self.assertEqual(f(txt, "<", ">"), ("b" , 3)) |