Add unit tests for regex expression

Author: iyanmv

test_regex.py

@@ -0,0 +1,66 @@
+import unittest
+import re
+
+with open('arxivbot.py', 'r') as f:
+    mod = f.readlines()
+    regex_expr = mod[6].split("'")[1]
+
+
+class TestRegexExpressionOneMatch(unittest.TestCase):
+    def setUp(self):
+        self.r = re.compile(regex_expr)
+        self.match = "https://arxiv.org/abs/1109.3195"
+    def test_single_url(self):
+        text = "https://arxiv.org/abs/1109.3195"
+        self.assertEqual(self.r.search(text).group(), self.match)
+    def test_single_url_period(self):
+        text = "https://arxiv.org/abs/1109.3195."
+        self.assertEqual(self.r.search(text).group(), self.match)
+    def test_single_url_inside_text(self):
+        text = "Hey! Check this paper: https://arxiv.org/abs/1109.3195. What do you think?"
+        self.assertEqual(self.r.search(text).group(), self.match)
+    def test_single_url_multiple_lines(self):
+        text = "\nhttps://arxiv.org/abs/1109.3195\n\n"
+        self.assertEqual(self.r.search(text).group(), self.match)
+    def test_single_url_parenthesis(self):
+        text = "This paper (https://arxiv.org/abs/1109.3195) looks interesting."
+        self.assertEqual(self.r.search(text).group(), self.match)
+
+class TestRegexExpressionOldURLs(unittest.TestCase):
+    def setUp(self):
+        self.r = re.compile(regex_expr)
+    def test_quant_ph(self):
+        text = "https://arxiv.org/abs/quant-ph/0512258"
+        self.assertEqual(self.r.search(text).group(), text)
+    def test_hep_ex(self):
+        text = "https://arxiv.org/abs/hep-ex/0102001"
+        self.assertEqual(self.r.search(text).group(), text)
+    def test_math_ph(self):
+        text = "https://arxiv.org/abs/math-ph/9810001"
+        self.assertEqual(self.r.search(text).group(), text)
+    def test_cs(self):
+        text = "https://arxiv.org/abs/cs/9902001"
+        self.assertEqual(self.r.search(text).group(), text)
+
+class TestRegexExpressionMultipleMatches(unittest.TestCase):
+    def setUp(self):
+        self.r = re.compile(regex_expr)
+        self.matches = ["https://arxiv.org/abs/quant-ph/0512258", "https://arxiv.org/abs/quant-ph/9806051",
+                        "https://arxiv.org/abs/0810.4372", "https://arxiv.org/abs/quant-ph/9810080"]
+    def test_multiple_matches_simple(self):
+        text = "https://arxiv.org/abs/quant-ph/0512258 https://arxiv.org/abs/quant-ph/9806051 "\
+             + "https://arxiv.org/abs/0810.4372 https://arxiv.org/abs/quant-ph/9810080"
+        self.assertEqual(self.r.findall(text), self.matches)
+    def test_multiple_matches_no_spaces(self):
+        text = "https://arxiv.org/abs/quant-ph/0512258https://arxiv.org/abs/quant-ph/9806051"\
+             + "https://arxiv.org/abs/0810.4372https://arxiv.org/abs/quant-ph/9810080"
+        self.assertEqual(self.r.findall(text), self.matches)
+    def test_multiple_matches_paragraph(self):
+        text = """https://arxiv.org/abs/quant-ph/0512258. Also, did you read this (https://arxiv.org/abs/quant-ph/9806051)?
+        And what about this other one: https://arxiv.org/abs/0810.4372
+        https://arxiv.org/abs/quant-ph/9810080"""
+        self.assertEqual(self.r.findall(text), self.matches)
+
+
+if __name__ == '__main__':
+    unittest.main()