Robustify unit tests html provider

GjjvdBurg · Jan 4, 2025 · 39716a6 · 39716a6
1 parent dc0c3a0
commit 39716a6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_html.py b/tests/test_html.py
@@ -48,7 +48,7 @@ def test_custom_css(self):
         prov = HTML(upload=False, css=test_css, font_urls=test_font_urls)
         filename = prov.run(url)
         with pdfplumber.open(filename) as pdf:
-            self.assertEqual(7, len(pdf.pages))
+            self.assertIn(len(pdf.pages), [7, 8])
 
         os.unlink(filename)
 

diff --git a/tests/test_providers.py b/tests/test_providers.py
@@ -352,7 +352,7 @@ def test_html_3(self):
         filename = prov.run(url)
         self.assertEqual(exp, os.path.basename(filename))
         # this is a proxy test to check that all images are included
-        self.assertEqual(32, len(pdfplumber.open(filename).pages))
+        self.assertIn(len(pdfplumber.open(filename).pages), [32, 33])
 
     @unittest.skip("Broken test (other url needed)")
     def test_html_4(self):