From 39716a6f18359fc542f14b1a68cbce95624d4963 Mon Sep 17 00:00:00 2001 From: Gertjan van den Burg Date: Sat, 4 Jan 2025 10:58:49 +0000 Subject: [PATCH] Robustify unit tests html provider --- tests/test_html.py | 2 +- tests/test_providers.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_html.py b/tests/test_html.py index cb32b51..ebdf4b4 100644 --- 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 index 6cb1996..0efb8f8 100644 --- 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):