def test_preference_adding_correctly(): user = User("Trump", "twitterking") movie=Movie("ooo",9999) movie.add_genre(Genre("onmygod")) user.add_preference(Genre("Action")) user.add_preference(Genre("Sci-Fi")) user.watch_movie(movie) ls=[Genre("onmygod"),Genre("Action"),Genre("Sci-Fi")] assert ls.sort()==recommend.get_user_preference(user).sort()
def test_preference_by_preference(): user = User("Trump", "twitterking") user.add_preference(Genre("Action")) user.add_preference(Genre("Sci-Fi")) assert user.preference.sort()==recommend.get_user_preference(user).sort() indicator = 0 for movie in recommend.get_recommended_watch_list(user): for genre in movie.genres: if genre in recommend.get_user_preference(user): indicator += 1 assert indicator > 0 # all movie is linked to preference indicator = 0
def test_multiple_way(): user = User("Trump", "twitterking") pref=[Genre("Crime")] for i in movie_list: user.watch_movie(i) for genre in i.genres: if genre not in pref: pref.append(genre) user.add_preference(Genre("Crime"))# not a genre of first 10 movie indicator = 0 indicator2 = 0 for movie in recommend.get_recommended_watch_list(user,10): #get 10 movie for genre in movie.genres: if genre in pref: indicator += 1 if genre == Genre("Crime"): indicator2 +=1 assert indicator > 0 # all movie is linked to preference assert indicator2 != 0 # def special(): # user = User("Trump", "twitterking") # pref = [Genre("Crime")] # for i in movie_list: # user.watch_movie(i) # for genre in i.genres: # if genre not in pref: # pref.append(genre) # user.add_preference(Genre("Crime")) # not a genre of first 10 movie # indicator = 0 # indicator2 = 0 # for movie in recommend.get_recommended_watch_list(user, 10): # get 10 movie # for genre in movie.genres: # if genre in pref: # indicator += 1 # if genre == Genre("Crime"): # indicator2 += 1 # return indicator2 != 0 # # f=0 # n=10000 # for i in range(n): # if special() == False: # f+=1 # print("{}%".format(f/n))