[tests] support one regex per URL for #pattern

2023-09-28 21:56:09 +02:00
parent be17103e21
commit bd3f7a5bbc
2 changed files with 25 additions and 4 deletions
--- a/test/test_results.py
+++ b/test/test_results.py
@@ -148,8 +148,13 @@ class TestExtractorResults(unittest.TestCase):

        if "#pattern" in result:
            self.assertGreater(len(tjob.url_list), 0)
-            for url in tjob.url_list:
-                self.assertRegex(url, result["#pattern"])
+            pattern = result["#pattern"]
+            if isinstance(pattern, str):
+                for url in tjob.url_list:
+                    self.assertRegex(url, pattern, msg="#pattern")
+            else:
+                for url, pat in zip(tjob.url_list, pattern):
+                    self.assertRegex(url, pat, msg="#pattern")

        if "#urls" in result:
            expected = result["#urls"]