diff --git a/hw5/auto_grader.py b/hw5/auto_grader.py index 8821005..1a31250 100644 --- a/hw5/auto_grader.py +++ b/hw5/auto_grader.py @@ -6,9 +6,13 @@ import json test_file = open('years.json','r') expected = json.loads(test_file.read()) -for l in range(80, 140, 20): +for l in range(80, 140, 10): m = MemoryLimitedHashMap(limit = l) actual = {k:v for k,v in Count(imdb_years(), m)} - print("Memory Limit", l, expected == actual) + + try: + assert(expected == actual) + except: + print("[#1] Failed for Memory Limit", l)