From 92754a7a12ec56dd958d879ececbc4f19e9623b0 Mon Sep 17 00:00:00 2001 From: Bryan Newbold Date: Sat, 8 Aug 2020 17:02:28 -0700 Subject: fix tests passing str as HTML --- python/tests/test_html.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'python/tests') diff --git a/python/tests/test_html.py b/python/tests/test_html.py index 3b59883..7d58a39 100644 --- a/python/tests/test_html.py +++ b/python/tests/test_html.py @@ -7,7 +7,7 @@ from sandcrawler.html import extract_fulltext_url def test_extract_fulltext_url(): - resp = extract_fulltext_url("asdf", "asdf") + resp = extract_fulltext_url("asdf", b"asdf") assert resp == {} resp = extract_fulltext_url( @@ -25,14 +25,14 @@ def test_extract_fulltext_url(): assert resp['pdf_url'] == "http://www.example.com/content/271/20/11761.full.pdf" assert resp['technique'] == "citation_pdf_url" - with open('tests/files/plos_one_article.html', 'r') as f: + with open('tests/files/plos_one_article.html', 'rb') as f: resp = extract_fulltext_url( "https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0213978", f.read(), ) assert resp['pdf_url'] == "https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0213978&type=printable" - with open('tests/files/elife_article.html', 'r') as f: + with open('tests/files/elife_article.html', 'rb') as f: resp = extract_fulltext_url( "https://elifesciences.org/articles/44753", f.read(), -- cgit v1.2.3