From 98956f19ed8016a435aae243a35ee2bbf5f8eba5 Mon Sep 17 00:00:00 2001 From: Eddie Hebert Date: Mon, 6 Jan 2014 10:52:32 -0500 Subject: [PATCH] MAINT: Make answer key reading compatible with Python 3. Ensure reading of bytes and checking against byte type, when parsing the Excel spreadsheet which contains the answers. --- tests/risk/answer_key.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/risk/answer_key.py b/tests/risk/answer_key.py index a1eb8ecf..50cce87e 100644 --- a/tests/risk/answer_key.py +++ b/tests/risk/answer_key.py @@ -51,12 +51,12 @@ LATEST_ANSWER_KEY_URL = ANSWER_KEY_DL_TEMPLATE.format( def answer_key_signature(): - with open(ANSWER_KEY_PATH, 'r') as f: + with open(ANSWER_KEY_PATH, 'rb') as f: md5 = hashlib.md5() - while True: + buf = f.read(1024) + md5.update(buf) + while buf != b"": buf = f.read(1024) - if not buf: - break md5.update(buf) return md5.hexdigest()