def test_actual_ratings3(self): probeFile = 'test/5_movies_1_cust/probe.txt' trainingSetDir = 'test/5_movies_1_cust/' movieIDDict = netflix_parse_train(trainingSetDir) self.assert_(movieIDDict == {'1': {'1': '1'}, '3': {'1': '1'}, '2': {'1': '1'}, '5': {'1': '1'}, '4': {'1': '1'}}) actualRatings = netflix_actual_ratings(probeFile, movieIDDict) self.assert_(actualRatings == ['1', '1', '1', '1', '1'])