fix tests

adbar · Nov 29, 2024 · 8b06b2d · 8b06b2d
1 parent 895cffe
commit 8b06b2d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/tests/cli_tests.py b/tests/cli_tests.py
@@ -510,7 +510,7 @@ def test_crawling():
     testargs = ["", "--crawl", ""]
     with patch.object(sys, "argv", testargs):
         args = cli.parse_args(testargs)
-        cli.process_args(args)
+    cli.process_args(args)
 
     testargs = ["", "--crawl", " "]
     with patch.object(sys, "argv", testargs):

diff --git a/tests/unit_tests.py b/tests/unit_tests.py
@@ -832,9 +832,12 @@ def test_extraction_options():
     # assert extract(my_html) is None
 
     # readability
+    my_html = '<html><body><p>' + 'Text. '*10 + '</p></body></html>'
+    result = etree.tostring(try_readability(html.fromstring(my_html)))
+    assert len(result) > 10 and b'Text' in result
     my_html = '<html><body><p>' + 'Text. '*10 + '<embed>Test</embed></p></body></html>'
     result = etree.tostring(try_readability(html.fromstring(my_html)))
-    assert len(result) > 10 and 'Test' not in result
+    assert b'Test' not in result
 
     my_html = '<html><head/><body>' + '<p>ABC def ghi jkl.</p>'*1000 + '<p>Posted on 1st Dec 2019<.</p></body></html>'
     assert bare_extraction(my_html, config=ZERO_CONFIG, with_metadata=True).date is not None