예제 #1
0
    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'])