diff --git a/test/test_results.py b/test/test_results.py index 119ac7f..74af29c 100644 --- a/test/test_results.py +++ b/test/test_results.py @@ -3,13 +3,12 @@ from app.filter import Filter from app.utils.session_utils import generate_user_keys from datetime import datetime from dateutil.parser import * -import json -import os def get_search_results(data): secret_key = generate_user_keys() - soup = Filter(user_keys=secret_key).clean(BeautifulSoup(data, 'html.parser')) + soup = Filter(user_keys=secret_key).clean( + BeautifulSoup(data, 'html.parser')) main_divs = soup.find('div', {'id': 'main'}) assert len(main_divs) > 1 @@ -17,7 +16,9 @@ def get_search_results(data): result_divs = [] for div in main_divs: # Result divs should only have 1 inner div - if len(list(div.children)) != 1 or not div.findChild() or 'div' not in div.findChild().name: + if (len(list(div.children)) != 1 + or not div.findChild() + or 'div' not in div.findChild().name): continue result_divs.append(div) @@ -78,6 +79,7 @@ def test_recent_results(client): try: date = parse(date_span) - assert (current_date - date).days <= (num_days + 5) # Date can have a little bit of wiggle room + # Date can have a little bit of wiggle room + assert (current_date - date).days <= (num_days + 5) except ParserError: pass