Python unit test for hw3 on dictionaries