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))
