[+] unittest for zmirror.utils.get_group()

Mticket · Sep 13, 2016 · 9d5bb4c · 9d5bb4c
1 parent cce3fb6
commit 9d5bb4c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_functions.py b/tests/test_functions.py
@@ -251,3 +251,15 @@ def test__encoding_detect(self):
         self.zmirror.possible_charsets = None
         self.zmirror.cchardet_available = False
         self.assertIsNone(self.zmirror.encoding_detect("测试中文".encode(encoding="utf-8")))
+
+    def test__get_group(self):
+        import re
+        m = re.match(r"""(?P<non_exist>\d+)?(?P<dog>dog)""", "dog")
+        self.assertEqual("", self.zmirror.get_group("non_exist", m))
+        self.assertEqual("", self.zmirror.get_group("cat", m))
+        self.assertEqual("dog", self.zmirror.get_group("dog", m))
+
+        m = re.match(r"""(?P<non_exist>\d+)?(?P<dog>dog)""", "cat")
+        self.assertEqual("", self.zmirror.get_group("non_exist", m))
+        self.assertEqual("", self.zmirror.get_group("cat", m))
+        self.assertEqual("", self.zmirror.get_group("dog", m))