def testAddFilmWithModeration(self):
added_film = AddedFilm(title='Test film',
localized_title='Test film PL',
release_year=2010,
user=self.u1)
added_film.save()
added_film.directors.add(self.p1)
added_film.directors.add(self.p2)
added_film.production_country.add(self.c1)
added_film.production_country.add(self.c2)
added_film.save()
AddedCharacter.objects.create(added_film=added_film,
person=self.p1,
character="Test character 1")
AddedCharacter.objects.create(added_film=added_film,
person=self.p2,
character="Test character 2")
AddedCharacter.objects.create(added_film=added_film,
person=self.p3,
character="Test character 3")
self.assertEqual(len(added_film.get_actors()), 3)
self.assertTrue(added_film.film is None)
self.assertEqual(added_film.moderation_status,
AddedFilm.STATUS_UNKNOWN)
added_film.accept(self.u2)
self.assertTrue(added_film.film is not None)
self.assertEqual(added_film.moderation_status,
AddedFilm.STATUS_ACCEPTED)
film = added_film.film
self.assertEqual(film.permalink, "test-film")
self.assertTrue(self.p1 in film.top_directors())
self.assertTrue(self.p2 in film.top_directors())
self.assertEqual(film.get_localized_title(), "Test film PL")
self.assertEqual(film.production_country_list, "USA,Poland")
self.assertEqual(len(film.get_actors()), 3)
def testAddFilmWithModeration( self ):
added_film = AddedFilm( title = 'Test film', localized_title='Test film PL',
release_year = 2010, user = self.u1 )
added_film.save()
added_film.directors.add( self.p1 )
added_film.directors.add( self.p2 )
added_film.production_country.add( self.c1 )
added_film.production_country.add( self.c2 )
added_film.save()
AddedCharacter.objects.create( added_film=added_film, person=self.p1, character="Test character 1" )
AddedCharacter.objects.create( added_film=added_film, person=self.p2, character="Test character 2" )
AddedCharacter.objects.create( added_film=added_film, person=self.p3, character="Test character 3" )
self.assertEqual( len( added_film.get_actors() ), 3 )
self.assertTrue( added_film.film is None )
self.assertEqual( added_film.moderation_status, AddedFilm.STATUS_UNKNOWN )
added_film.accept( self.u2 )
self.assertTrue( added_film.film is not None )
self.assertEqual( added_film.moderation_status, AddedFilm.STATUS_ACCEPTED )
film = added_film.film
self.assertEqual( film.permalink, "test-film" )
self.assertTrue( self.p1 in film.top_directors() )
self.assertTrue( self.p2 in film.top_directors() )
self.assertEqual( film.get_localized_title(), "Test film PL" )
self.assertEqual( film.production_country_list, "USA,Poland" )
self.assertEqual( len( film.get_actors() ), 3 )
def setUp(self):
User.objects.all().delete()
Object.objects.all().delete()
Film.objects.all().delete()
FilmLog.objects.all().delete()
Rating.objects.all().delete()
Recommendation.objects.all().delete()
Person.objects.all().delete()
AddedCharacter.objects.all().delete()
AddedFilm.objects.all().delete()
# set up users
self.u1 = User(username="******", email="*****@*****.**")
self.u1.save()
self.u2 = User.objects.create_user("root", "*****@*****.**", "root")
self.u2.user_permissions.add(Permission.objects.get(codename="can_accept_added_films"))
self.u2.save()
self.u3 = User(username="******", email="*****@*****.**")
self.u3.save()
self.u4 = User(username="******", email="*****@*****.**")
self.u4.save()
self.u5 = User(username="******", email="*****@*****.**")
self.u5.save()
self.u6 = User(username="******", email="*****@*****.**")
self.u6.save()
# some persons
self.p1 = Person(name="Clint", surname="Eastwood", imdb_code=None, type=Person.TYPE_PERSON)
self.p1.save()
self.p2 = Person(name="Sylvester", surname="Stallone", imdb_code=None, type=Person.TYPE_PERSON)
self.p2.save()
self.p3 = Person(name="Jack", surname="Mort", imdb_code=None, type=Person.TYPE_PERSON)
self.p3.save()
# set up films
self.f1 = Film(
type=1,
permalink="przypadek",
imdb_code=111,
status=1,
version=1,
release_year=1999,
title="Przypadek",
popularity=20,
popularity_month=10,
)
self.f1.save()
self.f2 = Film(
type=1,
permalink="wrestler",
imdb_code=112,
status=1,
version=1,
release_year=2008,
title="Wrestler",
popularity=10,
popularity_month=1,
)
self.f2.save()
self.f3 = Film(
type=1,
permalink="american-history-x",
imdb_code=113,
status=1,
version=1,
release_year=1998,
title="American History X",
popularity=1,
popularity_month=1,
)
self.f3.save()
self.f4 = Film(
type=1,
permalink="the-big-lebowski",
imdb_code=114,
status=1,
version=1,
release_year=1998,
title="The Big Lebowski",
popularity=1,
popularity_month=1,
)
self.f4.save()
self.f5 = Film(
type=1,
permalink="the-lord-of-the-rings-the-fellowship-of-the-ring",
imdb_code=115,
status=1,
version=1,
release_year=2001,
title="The Lord of the Rings: The Fellowship of the Ring",
popularity=1,
popularity_month=1,
)
self.f5.save()
self.f6 = Film(
type=1,
permalink="raiders-of-the-lost-ark",
imdb_code=116,
status=1,
version=1,
release_year=1981,
title="Raiders of the Lost Ark",
popularity=1,
popularity_month=1,
)
self.f6.save()
self.f7 = Film(
type=1,
permalink="the-alien",
imdb_code=117,
status=1,
version=1,
release_year=1979,
title="The Alien",
popularity=1,
popularity_month=1,
)
self.f7.save()
self.f8 = Film(
type=1,
permalink="terminator-2-judgment-day",
imdb_code=118,
status=1,
version=1,
release_year=1991,
title="Terminator 2: Judgment Day",
popularity=1,
popularity_month=1,
)
self.f8.save()
self.f9 = Film(
type=1,
permalink="przypadek",
imdb_code=119,
status=1,
version=1,
release_year=1999,
title="Przypadek",
popularity=20,
popularity_month=10,
)
self.f9.save()
self.f10 = Film(
type=1,
permalink="wrestler",
imdb_code=120,
status=1,
version=1,
release_year=2008,
title="Wrestler",
popularity=10,
popularity_month=1,
)
self.f10.save()
self.f11 = Film(
type=1,
permalink="american-history-x",
imdb_code=121,
status=1,
version=1,
release_year=1998,
title="American History X",
popularity=1,
popularity_month=1,
)
self.f11.save()
self.f12 = Film(
type=1,
permalink="the-big-lebowski",
imdb_code=122,
status=1,
version=1,
release_year=1998,
title="The Big Lebowski",
popularity=1,
popularity_month=1,
)
self.f12.save()
self.f13 = Film(
type=1,
permalink="the-lord-of-the-rings-the-fellowship-of-the-ring",
imdb_code=123,
status=1,
version=1,
release_year=2001,
title="The Lord of the Rings: The Fellowship of the Ring",
popularity=1,
popularity_month=1,
)
self.f13.save()
self.f14 = Film(
type=1,
permalink="raiders-of-the-lost-ark",
imdb_code=124,
status=1,
version=1,
release_year=1981,
title="Raiders of the Lost Ark",
popularity=1,
popularity_month=1,
)
self.f14.save()
self.f15 = Film(
type=1,
permalink="the-alien",
imdb_code=125,
status=1,
version=1,
release_year=1979,
title="The Alien",
popularity=1,
popularity_month=1,
)
self.f15.save()
self.f16 = Film(
type=1,
permalink="przypadek",
imdb_code=126,
status=1,
version=1,
release_year=1999,
title="Przypadek",
popularity=20,
popularity_month=10,
)
self.f16.save()
self.f17 = Film(
type=1,
permalink="wrestler",
imdb_code=127,
status=1,
version=1,
release_year=2008,
title="Wrestler",
popularity=10,
popularity_month=1,
)
self.f17.save()
self.f18 = Film(
type=1,
permalink="american-history-x",
imdb_code=128,
status=1,
version=1,
release_year=1998,
title="American History X",
popularity=1,
popularity_month=1,
)
self.f18.save()
self.f19 = Film(
type=1,
permalink="the-big-lebowski",
imdb_code=129,
status=1,
version=1,
release_year=1998,
title="The Big Lebowski",
popularity=1,
popularity_month=1,
)
self.f19.save()
self.f20 = Film(
type=1,
permalink="the-lord-of-the-rings-the-fellowship-of-the-ring",
imdb_code=130,
status=1,
version=1,
release_year=2001,
title="The Lord of the Rings: The Fellowship of the Ring",
popularity=1,
popularity_month=1,
)
self.f20.save()
self.f21 = Film(
type=1,
permalink="raiders-of-the-lost-ark",
imdb_code=131,
status=1,
version=1,
release_year=1981,
title="Raiders of the Lost Ark",
popularity=1,
popularity_month=1,
)
self.f21.save()
self.f22 = Film(
type=1,
permalink="the-alien",
imdb_code=132,
status=1,
version=1,
release_year=1979,
title="The Alien",
popularity=1,
popularity_month=1,
)
self.f22.save()
self.f23 = Film(
type=1,
permalink="terminator-2-judgment-day",
imdb_code=133,
status=1,
version=1,
release_year=1991,
title="Terminator 2: Judgment Day",
popularity=1,
popularity_month=1,
)
self.f23.save()
self.f24 = Film(
type=1,
permalink="przypadek",
imdb_code=134,
status=1,
version=1,
release_year=1999,
title="Przypadek",
popularity=20,
popularity_month=10,
)
self.f24.save()
self.f25 = Film(
type=1,
permalink="wrestler",
imdb_code=135,
status=1,
version=1,
release_year=2008,
title="Wrestler",
popularity=10,
popularity_month=1,
)
self.f25.save()
self.f26 = Film(
type=1,
permalink="american-history-x",
imdb_code=136,
status=1,
version=1,
release_year=1998,
title="American History X",
popularity=1,
popularity_month=1,
)
self.f26.save()
self.f27 = Film(
type=1,
permalink="the-big-lebowski",
imdb_code=137,
status=1,
version=1,
release_year=1998,
title="The Big Lebowski",
popularity=1,
popularity_month=1,
)
self.f27.save()
self.f28 = Film(
type=1,
permalink="the-lord-of-the-rings-the-fellowship-of-the-ring",
imdb_code=138,
status=1,
version=1,
release_year=2001,
title="The Lord of the Rings: The Fellowship of the Ring",
popularity=1,
popularity_month=1,
)
self.f28.save()
self.f29 = Film(
type=1,
permalink="raiders-of-the-lost-ark",
imdb_code=139,
status=1,
version=1,
release_year=1981,
title="Raiders of the Lost Ark",
popularity=1,
popularity_month=1,
)
self.f29.save()
self.f30 = Film(
type=1,
permalink="the-alien",
imdb_code=140,
status=1,
version=1,
release_year=1979,
title="The Alien",
popularity=1,
popularity_month=1,
)
self.f30.save()
self.f31 = AddedFilm(
title="Added film part 1", localized_title="Added film PL", release_year=2010, user=self.u1
)
self.f31.save()
self.f31.directors.add(self.p1)
self.f31.save()
AddedCharacter.objects.create(added_film=self.f31, person=self.p3, character="Jack Mort")
self.f31.accept(self.u2)
self.f32 = AddedFilm(
title="Added film part 2", localized_title="Added film PL", release_year=2010, user=self.u1
)
self.f32.save()
self.f32.directors.add(self.p2)
self.f32.save()
AddedCharacter.objects.create(added_film=self.f32, person=self.p3, character="Jack Mort")
self.f32.accept(self.u2)
# tagging
self.f1.save_tags("comedy drama horror animation mystery")
self.f2.save_tags("comedy sci-fi thriller series")
self.f3.save_tags("comedy")
self.f4.save_tags("sci-fi")
self.f5.save_tags("horror")
self.f6.save_tags("thriller")
self.f7.save_tags("drama")
self.f8.save_tags("series")
self.f9.save_tags("sci-fi")
self.f10.save_tags("fantasy")
self.f11.save_tags("mystery")
self.f12.save_tags("action")
self.f13.save_tags("adventure")
self.f14.save_tags("animation")
self.f15.save_tags("family")
self.f16.save_tags("thriller")
self.f17.save_tags("drama")
self.f18.save_tags("series")
self.f19.save_tags("sci-fi")
self.f20.save_tags("fantasy")
self.f21.save_tags("comedy")
self.f22.save_tags("animation")
self.f23.save_tags("comedy")
self.f24.save_tags("sci-fi")
self.f25.save_tags("horror")
self.f26.save_tags("thriller")
self.f27.save_tags("drama")
self.f28.save_tags("series")
self.f29.save_tags("sci-fi")
self.f30.save_tags("fantasy")
# First user ratings
save_rating(self.f1, self.u1, 1)
save_rating(self.f2, self.u1, 2)
save_rating(self.f3, self.u1, 3)
save_rating(self.f4, self.u1, 4)
save_rating(self.f5, self.u1, 5)
save_rating(self.f6, self.u1, 6)
save_rating(self.f7, self.u1, 7)
save_rating(self.f8, self.u1, 8)
save_rating(self.f9, self.u1, 9)
save_rating(self.f10, self.u1, 10)
save_rating(self.f11, self.u1, 1)
save_rating(self.f12, self.u1, 2)
save_rating(self.f13, self.u1, 3)
save_rating(self.f14, self.u1, 4)
save_rating(self.f15, self.u1, 5)
save_rating(self.f16, self.u1, 6)
save_rating(self.f17, self.u1, 7)
save_rating(self.f18, self.u1, 8)
save_rating(self.f19, self.u1, 9)
save_rating(self.f20, self.u1, 10)
save_rating(self.f21, self.u1, 1)
save_rating(self.f22, self.u1, 2)
save_rating(self.f23, self.u1, 3)
save_rating(self.f24, self.u1, 4)
save_rating(self.f25, self.u1, 5)
save_rating(self.f26, self.u1, 6)
save_rating(self.f27, self.u1, 7)
save_rating(self.f28, self.u1, 8)
save_rating(self.f29, self.u1, 9)
save_rating(self.f30, self.u1, 10)
# Second user ratings - more than minimal common films requirement (15)
save_rating(self.f1, self.u2, 1)
save_rating(self.f2, self.u2, 2)
save_rating(self.f3, self.u2, 3)
save_rating(self.f4, self.u2, 4)
save_rating(self.f5, self.u2, 5)
save_rating(self.f6, self.u2, 6)
save_rating(self.f7, self.u2, 7)
save_rating(self.f8, self.u2, 8)
save_rating(self.f9, self.u2, 9)
save_rating(self.f10, self.u2, 10)
save_rating(self.f11, self.u2, 1)
# save_rating(self.f12, self.u2, 2)
save_rating(self.f13, self.u2, 3)
# save_rating(self.f14, self.u2, 4)
save_rating(self.f15, self.u2, 5)
# save_rating(self.f16, self.u2, 6)
save_rating(self.f17, self.u2, 7)
# save_rating(self.f18, self.u2, 8)
save_rating(self.f19, self.u2, 9)
# save_rating(self.f20, self.u2, 10)
save_rating(self.f21, self.u2, 1)
# save_rating(self.f22, self.u2, 2)
save_rating(self.f23, self.u2, 3)
# save_rating(self.f24, self.u2, 4)
save_rating(self.f25, self.u2, 5)
# save_rating(self.f26, self.u2, 6)
save_rating(self.f27, self.u2, 7)
# save_rating(self.f28, self.u2, 8)
save_rating(self.f29, self.u2, 9)
# save_rating(self.f30, self.u2, 10)
# Third user ratings - equal with minimal common films requirement (15)
# save_rating(self.f1, self.u3, 1)
# save_rating(self.f2, self.u3, 2)
# save_rating(self.f3, self.u3, 3)
# save_rating(self.f4, self.u3, 4)
# save_rating(self.f5, self.u3, 5)
# save_rating(self.f6, self.u3, 6)
# save_rating(self.f7, self.u3, 7)
# save_rating(self.f8, self.u3, 8)
# save_rating(self.f9, self.u3, 9)
# save_rating(self.f10, self.u3, 10)
# save_rating(self.f11, self.u3, 1)
# save_rating(self.f12, self.u3, 2)
# save_rating(self.f13, self.u3, 3)
# save_rating(self.f14, self.u3, 4)
# save_rating(self.f15, self.u3, 5)
save_rating(self.f16, self.u3, 6)
save_rating(self.f17, self.u3, 7)
save_rating(self.f18, self.u3, 8)
save_rating(self.f19, self.u3, 9)
save_rating(self.f20, self.u3, 10)
save_rating(self.f21, self.u3, 1)
save_rating(self.f22, self.u3, 2)
save_rating(self.f23, self.u3, 3)
save_rating(self.f24, self.u3, 4)
save_rating(self.f25, self.u3, 5)
save_rating(self.f26, self.u3, 6)
save_rating(self.f27, self.u3, 7)
save_rating(self.f28, self.u3, 8)
save_rating(self.f29, self.u3, 9)
save_rating(self.f30, self.u3, 10)
# Fourth user ratings - less than minimal films requirement
# save_rating(self.f1, self.u4, 1)
# save_rating(self.f2, self.u4, 2)
# save_rating(self.f3, self.u4, 3)
# save_rating(self.f4, self.u4, 4)
# save_rating(self.f5, self.u4, 5)
# save_rating(self.f6, self.u4, 6)
# save_rating(self.f7, self.u4, 7)
# save_rating(self.f8, self.u4, 8)
# save_rating(self.f9, self.u4, 9)
# save_rating(self.f10, self.u4, 10)
# save_rating(self.f11, self.u4, 1)
# save_rating(self.f12, self.u4, 2)
# save_rating(self.f13, self.u4, 3)
# save_rating(self.f14, self.u4, 4)
# save_rating(self.f15, self.u4, 5)
# save_rating(self.f16, self.u4, 6)
# save_rating(self.f17, self.u4, 7)
# save_rating(self.f18, self.u4, 8)
# save_rating(self.f19, self.u4, 9)
# save_rating(self.f20, self.u4, 10)
save_rating(self.f21, self.u4, 1)
save_rating(self.f22, self.u4, 2)
save_rating(self.f23, self.u4, 3)
save_rating(self.f24, self.u4, 4)
save_rating(self.f25, self.u4, 5)
save_rating(self.f26, self.u4, 6)
save_rating(self.f27, self.u4, 7)
save_rating(self.f28, self.u4, 8)
save_rating(self.f29, self.u4, 9)
save_rating(self.f30, self.u4, 10)
# Fifth user ratings - the same films rated (1st and 5th user) but different
# ratings
# save_rating(self.f1, self.u5, 10)
# save_rating(self.f2, self.u5, 9)
# save_rating(self.f3, self.u5, 8)
# save_rating(self.f4, self.u5, 7)
# save_rating(self.f5, self.u5, 6)
# save_rating(self.f6, self.u5, 5)
# save_rating(self.f7, self.u5, 4)
# save_rating(self.f8, self.u5, 3)
# save_rating(self.f9, self.u5, 2)
# save_rating(self.f10, self.u5, 1)
# save_rating(self.f11, self.u5, 10)
# save_rating(self.f12, self.u5, 9)
# save_rating(self.f13, self.u5, 8)
# save_rating(self.f14, self.u5, 7)
# save_rating(self.f15, self.u5, 6)
# save_rating(self.f16, self.u5, 5)
# save_rating(self.f17, self.u5, 4)
# save_rating(self.f18, self.u5, 3)
# save_rating(self.f19, self.u5, 2)
# save_rating(self.f20, self.u5, 1)
# save_rating(self.f21, self.u5, 10)
# save_rating(self.f22, self.u5, 9)
# save_rating(self.f23, self.u5, 8)
# save_rating(self.f24, self.u5, 7)
# save_rating(self.f25, self.u5, 6)
# save_rating(self.f26, self.u5, 5)
# save_rating(self.f27, self.u5, 4)
# save_rating(self.f28, self.u5, 3)
# save_rating(self.f29, self.u5, 2)
# save_rating(self.f30, self.u5, 1)
# Sixth user ratings - the same films rated (1st and 5th user) but different
# ratings
save_rating(self.f1, self.u6, 10)
save_rating(self.f2, self.u6, 10)
save_rating(self.f3, self.u6, 10)
save_rating(self.f4, self.u6, 10)
save_rating(self.f5, self.u6, 10)
save_rating(self.f6, self.u6, 10)
save_rating(self.f7, self.u6, 10)
save_rating(self.f8, self.u6, 10)
save_rating(self.f9, self.u6, 10)
save_rating(self.f10, self.u6, 10)
save_rating(self.f11, self.u6, 10)
save_rating(self.f12, self.u6, 10)
save_rating(self.f13, self.u6, 10)
save_rating(self.f14, self.u6, 10)
save_rating(self.f15, self.u6, 10)
save_rating(self.f16, self.u6, 10)
save_rating(self.f17, self.u6, 10)
save_rating(self.f18, self.u6, 10)
save_rating(self.f19, self.u6, 10)
save_rating(self.f20, self.u6, 10)
save_rating(self.f21, self.u6, 10)
save_rating(self.f22, self.u6, 10)
save_rating(self.f23, self.u6, 10)
save_rating(self.f24, self.u6, 10)
save_rating(self.f25, self.u6, 10)
save_rating(self.f26, self.u6, 10)
save_rating(self.f27, self.u6, 10)
save_rating(self.f28, self.u6, 10)
save_rating(self.f29, self.u6, 10)
save_rating(self.f30, self.u6, 10)
class QueriesTestCase(TestCase):
def setUp(self):
User.objects.all().delete()
Object.objects.all().delete()
Film.objects.all().delete()
FilmLog.objects.all().delete()
Rating.objects.all().delete()
Recommendation.objects.all().delete()
Person.objects.all().delete()
AddedCharacter.objects.all().delete()
AddedFilm.objects.all().delete()
# set up users
self.u1 = User(username="******", email="*****@*****.**")
self.u1.save()
self.u2 = User.objects.create_user("root", "*****@*****.**", "root")
self.u2.user_permissions.add(Permission.objects.get(codename="can_accept_added_films"))
self.u2.save()
self.u3 = User(username="******", email="*****@*****.**")
self.u3.save()
self.u4 = User(username="******", email="*****@*****.**")
self.u4.save()
self.u5 = User(username="******", email="*****@*****.**")
self.u5.save()
self.u6 = User(username="******", email="*****@*****.**")
self.u6.save()
# some persons
self.p1 = Person(name="Clint", surname="Eastwood", imdb_code=None, type=Person.TYPE_PERSON)
self.p1.save()
self.p2 = Person(name="Sylvester", surname="Stallone", imdb_code=None, type=Person.TYPE_PERSON)
self.p2.save()
self.p3 = Person(name="Jack", surname="Mort", imdb_code=None, type=Person.TYPE_PERSON)
self.p3.save()
# set up films
self.f1 = Film(
type=1,
permalink="przypadek",
imdb_code=111,
status=1,
version=1,
release_year=1999,
title="Przypadek",
popularity=20,
popularity_month=10,
)
self.f1.save()
self.f2 = Film(
type=1,
permalink="wrestler",
imdb_code=112,
status=1,
version=1,
release_year=2008,
title="Wrestler",
popularity=10,
popularity_month=1,
)
self.f2.save()
self.f3 = Film(
type=1,
permalink="american-history-x",
imdb_code=113,
status=1,
version=1,
release_year=1998,
title="American History X",
popularity=1,
popularity_month=1,
)
self.f3.save()
self.f4 = Film(
type=1,
permalink="the-big-lebowski",
imdb_code=114,
status=1,
version=1,
release_year=1998,
title="The Big Lebowski",
popularity=1,
popularity_month=1,
)
self.f4.save()
self.f5 = Film(
type=1,
permalink="the-lord-of-the-rings-the-fellowship-of-the-ring",
imdb_code=115,
status=1,
version=1,
release_year=2001,
title="The Lord of the Rings: The Fellowship of the Ring",
popularity=1,
popularity_month=1,
)
self.f5.save()
self.f6 = Film(
type=1,
permalink="raiders-of-the-lost-ark",
imdb_code=116,
status=1,
version=1,
release_year=1981,
title="Raiders of the Lost Ark",
popularity=1,
popularity_month=1,
)
self.f6.save()
self.f7 = Film(
type=1,
permalink="the-alien",
imdb_code=117,
status=1,
version=1,
release_year=1979,
title="The Alien",
popularity=1,
popularity_month=1,
)
self.f7.save()
self.f8 = Film(
type=1,
permalink="terminator-2-judgment-day",
imdb_code=118,
status=1,
version=1,
release_year=1991,
title="Terminator 2: Judgment Day",
popularity=1,
popularity_month=1,
)
self.f8.save()
self.f9 = Film(
type=1,
permalink="przypadek",
imdb_code=119,
status=1,
version=1,
release_year=1999,
title="Przypadek",
popularity=20,
popularity_month=10,
)
self.f9.save()
self.f10 = Film(
type=1,
permalink="wrestler",
imdb_code=120,
status=1,
version=1,
release_year=2008,
title="Wrestler",
popularity=10,
popularity_month=1,
)
self.f10.save()
self.f11 = Film(
type=1,
permalink="american-history-x",
imdb_code=121,
status=1,
version=1,
release_year=1998,
title="American History X",
popularity=1,
popularity_month=1,
)
self.f11.save()
self.f12 = Film(
type=1,
permalink="the-big-lebowski",
imdb_code=122,
status=1,
version=1,
release_year=1998,
title="The Big Lebowski",
popularity=1,
popularity_month=1,
)
self.f12.save()
self.f13 = Film(
type=1,
permalink="the-lord-of-the-rings-the-fellowship-of-the-ring",
imdb_code=123,
status=1,
version=1,
release_year=2001,
title="The Lord of the Rings: The Fellowship of the Ring",
popularity=1,
popularity_month=1,
)
self.f13.save()
self.f14 = Film(
type=1,
permalink="raiders-of-the-lost-ark",
imdb_code=124,
status=1,
version=1,
release_year=1981,
title="Raiders of the Lost Ark",
popularity=1,
popularity_month=1,
)
self.f14.save()
self.f15 = Film(
type=1,
permalink="the-alien",
imdb_code=125,
status=1,
version=1,
release_year=1979,
title="The Alien",
popularity=1,
popularity_month=1,
)
self.f15.save()
self.f16 = Film(
type=1,
permalink="przypadek",
imdb_code=126,
status=1,
version=1,
release_year=1999,
title="Przypadek",
popularity=20,
popularity_month=10,
)
self.f16.save()
self.f17 = Film(
type=1,
permalink="wrestler",
imdb_code=127,
status=1,
version=1,
release_year=2008,
title="Wrestler",
popularity=10,
popularity_month=1,
)
self.f17.save()
self.f18 = Film(
type=1,
permalink="american-history-x",
imdb_code=128,
status=1,
version=1,
release_year=1998,
title="American History X",
popularity=1,
popularity_month=1,
)
self.f18.save()
self.f19 = Film(
type=1,
permalink="the-big-lebowski",
imdb_code=129,
status=1,
version=1,
release_year=1998,
title="The Big Lebowski",
popularity=1,
popularity_month=1,
)
self.f19.save()
self.f20 = Film(
type=1,
permalink="the-lord-of-the-rings-the-fellowship-of-the-ring",
imdb_code=130,
status=1,
version=1,
release_year=2001,
title="The Lord of the Rings: The Fellowship of the Ring",
popularity=1,
popularity_month=1,
)
self.f20.save()
self.f21 = Film(
type=1,
permalink="raiders-of-the-lost-ark",
imdb_code=131,
status=1,
version=1,
release_year=1981,
title="Raiders of the Lost Ark",
popularity=1,
popularity_month=1,
)
self.f21.save()
self.f22 = Film(
type=1,
permalink="the-alien",
imdb_code=132,
status=1,
version=1,
release_year=1979,
title="The Alien",
popularity=1,
popularity_month=1,
)
self.f22.save()
self.f23 = Film(
type=1,
permalink="terminator-2-judgment-day",
imdb_code=133,
status=1,
version=1,
release_year=1991,
title="Terminator 2: Judgment Day",
popularity=1,
popularity_month=1,
)
self.f23.save()
self.f24 = Film(
type=1,
permalink="przypadek",
imdb_code=134,
status=1,
version=1,
release_year=1999,
title="Przypadek",
popularity=20,
popularity_month=10,
)
self.f24.save()
self.f25 = Film(
type=1,
permalink="wrestler",
imdb_code=135,
status=1,
version=1,
release_year=2008,
title="Wrestler",
popularity=10,
popularity_month=1,
)
self.f25.save()
self.f26 = Film(
type=1,
permalink="american-history-x",
imdb_code=136,
status=1,
version=1,
release_year=1998,
title="American History X",
popularity=1,
popularity_month=1,
)
self.f26.save()
self.f27 = Film(
type=1,
permalink="the-big-lebowski",
imdb_code=137,
status=1,
version=1,
release_year=1998,
title="The Big Lebowski",
popularity=1,
popularity_month=1,
)
self.f27.save()
self.f28 = Film(
type=1,
permalink="the-lord-of-the-rings-the-fellowship-of-the-ring",
imdb_code=138,
status=1,
version=1,
release_year=2001,
title="The Lord of the Rings: The Fellowship of the Ring",
popularity=1,
popularity_month=1,
)
self.f28.save()
self.f29 = Film(
type=1,
permalink="raiders-of-the-lost-ark",
imdb_code=139,
status=1,
version=1,
release_year=1981,
title="Raiders of the Lost Ark",
popularity=1,
popularity_month=1,
)
self.f29.save()
self.f30 = Film(
type=1,
permalink="the-alien",
imdb_code=140,
status=1,
version=1,
release_year=1979,
title="The Alien",
popularity=1,
popularity_month=1,
)
self.f30.save()
self.f31 = AddedFilm(
title="Added film part 1", localized_title="Added film PL", release_year=2010, user=self.u1
)
self.f31.save()
self.f31.directors.add(self.p1)
self.f31.save()
AddedCharacter.objects.create(added_film=self.f31, person=self.p3, character="Jack Mort")
self.f31.accept(self.u2)
self.f32 = AddedFilm(
title="Added film part 2", localized_title="Added film PL", release_year=2010, user=self.u1
)
self.f32.save()
self.f32.directors.add(self.p2)
self.f32.save()
AddedCharacter.objects.create(added_film=self.f32, person=self.p3, character="Jack Mort")
self.f32.accept(self.u2)
# tagging
self.f1.save_tags("comedy drama horror animation mystery")
self.f2.save_tags("comedy sci-fi thriller series")
self.f3.save_tags("comedy")
self.f4.save_tags("sci-fi")
self.f5.save_tags("horror")
self.f6.save_tags("thriller")
self.f7.save_tags("drama")
self.f8.save_tags("series")
self.f9.save_tags("sci-fi")
self.f10.save_tags("fantasy")
self.f11.save_tags("mystery")
self.f12.save_tags("action")
self.f13.save_tags("adventure")
self.f14.save_tags("animation")
self.f15.save_tags("family")
self.f16.save_tags("thriller")
self.f17.save_tags("drama")
self.f18.save_tags("series")
self.f19.save_tags("sci-fi")
self.f20.save_tags("fantasy")
self.f21.save_tags("comedy")
self.f22.save_tags("animation")
self.f23.save_tags("comedy")
self.f24.save_tags("sci-fi")
self.f25.save_tags("horror")
self.f26.save_tags("thriller")
self.f27.save_tags("drama")
self.f28.save_tags("series")
self.f29.save_tags("sci-fi")
self.f30.save_tags("fantasy")
# First user ratings
save_rating(self.f1, self.u1, 1)
save_rating(self.f2, self.u1, 2)
save_rating(self.f3, self.u1, 3)
save_rating(self.f4, self.u1, 4)
save_rating(self.f5, self.u1, 5)
save_rating(self.f6, self.u1, 6)
save_rating(self.f7, self.u1, 7)
save_rating(self.f8, self.u1, 8)
save_rating(self.f9, self.u1, 9)
save_rating(self.f10, self.u1, 10)
save_rating(self.f11, self.u1, 1)
save_rating(self.f12, self.u1, 2)
save_rating(self.f13, self.u1, 3)
save_rating(self.f14, self.u1, 4)
save_rating(self.f15, self.u1, 5)
save_rating(self.f16, self.u1, 6)
save_rating(self.f17, self.u1, 7)
save_rating(self.f18, self.u1, 8)
save_rating(self.f19, self.u1, 9)
save_rating(self.f20, self.u1, 10)
save_rating(self.f21, self.u1, 1)
save_rating(self.f22, self.u1, 2)
save_rating(self.f23, self.u1, 3)
save_rating(self.f24, self.u1, 4)
save_rating(self.f25, self.u1, 5)
save_rating(self.f26, self.u1, 6)
save_rating(self.f27, self.u1, 7)
save_rating(self.f28, self.u1, 8)
save_rating(self.f29, self.u1, 9)
save_rating(self.f30, self.u1, 10)
# Second user ratings - more than minimal common films requirement (15)
save_rating(self.f1, self.u2, 1)
save_rating(self.f2, self.u2, 2)
save_rating(self.f3, self.u2, 3)
save_rating(self.f4, self.u2, 4)
save_rating(self.f5, self.u2, 5)
save_rating(self.f6, self.u2, 6)
save_rating(self.f7, self.u2, 7)
save_rating(self.f8, self.u2, 8)
save_rating(self.f9, self.u2, 9)
save_rating(self.f10, self.u2, 10)
save_rating(self.f11, self.u2, 1)
# save_rating(self.f12, self.u2, 2)
save_rating(self.f13, self.u2, 3)
# save_rating(self.f14, self.u2, 4)
save_rating(self.f15, self.u2, 5)
# save_rating(self.f16, self.u2, 6)
save_rating(self.f17, self.u2, 7)
# save_rating(self.f18, self.u2, 8)
save_rating(self.f19, self.u2, 9)
# save_rating(self.f20, self.u2, 10)
save_rating(self.f21, self.u2, 1)
# save_rating(self.f22, self.u2, 2)
save_rating(self.f23, self.u2, 3)
# save_rating(self.f24, self.u2, 4)
save_rating(self.f25, self.u2, 5)
# save_rating(self.f26, self.u2, 6)
save_rating(self.f27, self.u2, 7)
# save_rating(self.f28, self.u2, 8)
save_rating(self.f29, self.u2, 9)
# save_rating(self.f30, self.u2, 10)
# Third user ratings - equal with minimal common films requirement (15)
# save_rating(self.f1, self.u3, 1)
# save_rating(self.f2, self.u3, 2)
# save_rating(self.f3, self.u3, 3)
# save_rating(self.f4, self.u3, 4)
# save_rating(self.f5, self.u3, 5)
# save_rating(self.f6, self.u3, 6)
# save_rating(self.f7, self.u3, 7)
# save_rating(self.f8, self.u3, 8)
# save_rating(self.f9, self.u3, 9)
# save_rating(self.f10, self.u3, 10)
# save_rating(self.f11, self.u3, 1)
# save_rating(self.f12, self.u3, 2)
# save_rating(self.f13, self.u3, 3)
# save_rating(self.f14, self.u3, 4)
# save_rating(self.f15, self.u3, 5)
save_rating(self.f16, self.u3, 6)
save_rating(self.f17, self.u3, 7)
save_rating(self.f18, self.u3, 8)
save_rating(self.f19, self.u3, 9)
save_rating(self.f20, self.u3, 10)
save_rating(self.f21, self.u3, 1)
save_rating(self.f22, self.u3, 2)
save_rating(self.f23, self.u3, 3)
save_rating(self.f24, self.u3, 4)
save_rating(self.f25, self.u3, 5)
save_rating(self.f26, self.u3, 6)
save_rating(self.f27, self.u3, 7)
save_rating(self.f28, self.u3, 8)
save_rating(self.f29, self.u3, 9)
save_rating(self.f30, self.u3, 10)
# Fourth user ratings - less than minimal films requirement
# save_rating(self.f1, self.u4, 1)
# save_rating(self.f2, self.u4, 2)
# save_rating(self.f3, self.u4, 3)
# save_rating(self.f4, self.u4, 4)
# save_rating(self.f5, self.u4, 5)
# save_rating(self.f6, self.u4, 6)
# save_rating(self.f7, self.u4, 7)
# save_rating(self.f8, self.u4, 8)
# save_rating(self.f9, self.u4, 9)
# save_rating(self.f10, self.u4, 10)
# save_rating(self.f11, self.u4, 1)
# save_rating(self.f12, self.u4, 2)
# save_rating(self.f13, self.u4, 3)
# save_rating(self.f14, self.u4, 4)
# save_rating(self.f15, self.u4, 5)
# save_rating(self.f16, self.u4, 6)
# save_rating(self.f17, self.u4, 7)
# save_rating(self.f18, self.u4, 8)
# save_rating(self.f19, self.u4, 9)
# save_rating(self.f20, self.u4, 10)
save_rating(self.f21, self.u4, 1)
save_rating(self.f22, self.u4, 2)
save_rating(self.f23, self.u4, 3)
save_rating(self.f24, self.u4, 4)
save_rating(self.f25, self.u4, 5)
save_rating(self.f26, self.u4, 6)
save_rating(self.f27, self.u4, 7)
save_rating(self.f28, self.u4, 8)
save_rating(self.f29, self.u4, 9)
save_rating(self.f30, self.u4, 10)
# Fifth user ratings - the same films rated (1st and 5th user) but different
# ratings
# save_rating(self.f1, self.u5, 10)
# save_rating(self.f2, self.u5, 9)
# save_rating(self.f3, self.u5, 8)
# save_rating(self.f4, self.u5, 7)
# save_rating(self.f5, self.u5, 6)
# save_rating(self.f6, self.u5, 5)
# save_rating(self.f7, self.u5, 4)
# save_rating(self.f8, self.u5, 3)
# save_rating(self.f9, self.u5, 2)
# save_rating(self.f10, self.u5, 1)
# save_rating(self.f11, self.u5, 10)
# save_rating(self.f12, self.u5, 9)
# save_rating(self.f13, self.u5, 8)
# save_rating(self.f14, self.u5, 7)
# save_rating(self.f15, self.u5, 6)
# save_rating(self.f16, self.u5, 5)
# save_rating(self.f17, self.u5, 4)
# save_rating(self.f18, self.u5, 3)
# save_rating(self.f19, self.u5, 2)
# save_rating(self.f20, self.u5, 1)
# save_rating(self.f21, self.u5, 10)
# save_rating(self.f22, self.u5, 9)
# save_rating(self.f23, self.u5, 8)
# save_rating(self.f24, self.u5, 7)
# save_rating(self.f25, self.u5, 6)
# save_rating(self.f26, self.u5, 5)
# save_rating(self.f27, self.u5, 4)
# save_rating(self.f28, self.u5, 3)
# save_rating(self.f29, self.u5, 2)
# save_rating(self.f30, self.u5, 1)
# Sixth user ratings - the same films rated (1st and 5th user) but different
# ratings
save_rating(self.f1, self.u6, 10)
save_rating(self.f2, self.u6, 10)
save_rating(self.f3, self.u6, 10)
save_rating(self.f4, self.u6, 10)
save_rating(self.f5, self.u6, 10)
save_rating(self.f6, self.u6, 10)
save_rating(self.f7, self.u6, 10)
save_rating(self.f8, self.u6, 10)
save_rating(self.f9, self.u6, 10)
save_rating(self.f10, self.u6, 10)
save_rating(self.f11, self.u6, 10)
save_rating(self.f12, self.u6, 10)
save_rating(self.f13, self.u6, 10)
save_rating(self.f14, self.u6, 10)
save_rating(self.f15, self.u6, 10)
save_rating(self.f16, self.u6, 10)
save_rating(self.f17, self.u6, 10)
save_rating(self.f18, self.u6, 10)
save_rating(self.f19, self.u6, 10)
save_rating(self.f20, self.u6, 10)
save_rating(self.f21, self.u6, 10)
save_rating(self.f22, self.u6, 10)
save_rating(self.f23, self.u6, 10)
save_rating(self.f24, self.u6, 10)
save_rating(self.f25, self.u6, 10)
save_rating(self.f26, self.u6, 10)
save_rating(self.f27, self.u6, 10)
save_rating(self.f28, self.u6, 10)
save_rating(self.f29, self.u6, 10)
save_rating(self.f30, self.u6, 10)
@unittest.skipIf(not is_postgres, "sqlite is not supported yet")
def test_get_recently_popular_films_query(self):
print "Test get recently popular films query"
"""
Test for query which gets the most popular films for the rating machine.
First it should return all films, then we rate one and in the second call
it should only retrieve the other film.
"""
recom_helper = RecomHelper()
film_helper = FilmHelper()
u1_result = recom_helper.get_recently_popular_films_query(self.u1.id)
self.assertEquals(len(u1_result), 2)
u2_result = recom_helper.get_recently_popular_films_query(self.u2.id)
self.assertEquals(len(u2_result), 12)
film_with_director_and_actor = Film.objects.filter(title__startswith="Added film part 1")
related_directors = film_helper.get_film_directors(film_with_director_and_actor[0])
related_characters = film_helper.get_film_actors(film_with_director_and_actor[0])
related_actors = []
for character in related_characters:
actor = character.person
related_actors.append(actor)
POST = {}
# TODO check why it starts breaking (form invalid) when form_id is set
# POST['form_id'] = '1'
POST["rating"] = "6"
POST["object_id"] = str(self.f12.parent.id)
POST["form_type"] = str(Rating.TYPE_FILM)
POST["cancel_rating"] = "0"
rating_form = RatingForm(POST)
if rating_form.is_valid():
rating_helper.add_edit_rating(rating_form, self.u2)
else:
print "RATING FORM INVALID!"
result = recom_helper.get_recently_popular_films_query(self.u2.id)
self.assertEquals(len(result), 11)
# test filtering
result = recom_helper.get_recently_popular_films_query(self.u5.id)
self.assertEquals(len(result), 32)
result = recom_helper.get_recently_popular_films_query(self.u5.id, year_from="1990")
self.assertEquals(len(result), 24)
result = recom_helper.get_recently_popular_films_query(self.u5.id, year_to="2000")
self.assertEquals(len(result), 22)
result = recom_helper.get_recently_popular_films_query(self.u5.id, year_from="1990", year_to="2000")
self.assertEquals(len(result), 14)
result = recom_helper.get_recently_popular_films_query(self.u5.id, popularity="10")
self.assertEquals(len(result), 8)
result = recom_helper.get_recently_popular_films_query(
self.u5.id, year_from="1990", year_to="2000", popularity="10"
)
self.assertEquals(len(result), 4)
rated_films = [self.f1.id, self.f3.id, self.f5.id, self.f7.id, self.f9.id]
result = recom_helper.get_recently_popular_films_query(self.u5.id, rated_films=rated_films)
self.assertEquals(len(result), 27)
result = recom_helper.get_recently_popular_films_query(
self.u5.id, year_from="1990", year_to="2000", rated_films=rated_films, popularity="10"
)
self.assertEquals(len(result), 2)
result = recom_helper.get_recently_popular_films_query(self.u5.id, related_director=related_directors)
self.assertEquals(len(result), 1)
result = recom_helper.get_recently_popular_films_query(self.u5.id, related_actor=related_actors)
self.assertEquals(len(result), 2)
result = recom_helper.get_recently_popular_films_query(
self.u5.id, related_director=related_directors, related_actor=related_actors
)
self.assertEquals(len(result), 1)
result = recom_helper.get_recently_popular_films_query(
self.u5.id,
year_from="1990",
year_to="2011",
related_director=related_directors,
related_actor=related_actors,
)
self.assertEquals(len(result), 1)
@unittest.skipIf(not is_postgres, "postgres dependend")
def test_get_best_psi_films_queryset_alg2(self):
"""
Test for query that retrieves top recommendations.
First it should return all films, then we rate one and in the second call
it should only retrieve the other film.
"""
recom_helper = RecomHelper()
film_helper = FilmHelper()
# no recommendations yet, the result should be 0
result = recom_helper.get_best_psi_films_queryset_alg2(self.u4.id)
self.assertEquals(len(result), 0)
# simulate generating recommendations (for all films) and test again, it should retrieve all films
r1 = Recommendation()
r1.guess_rating_alg2 = 7
r1.film = self.f19
r1.user = self.u4
r1.save()
r2 = Recommendation()
r2.guess_rating_alg2 = 8
r2.film = self.f20
r2.user = self.u4
r2.save()
result = recom_helper.get_best_psi_films_queryset_alg2(self.u4.id)
self.assertEquals(len(result), 2)
# rate one film, it should be removed from the list
POST = {}
# TODO check why it starts breaking (form invalid) when form_id is set
# POST['form_id'] = '1'
POST["rating"] = "6"
POST["object_id"] = str(self.f20.parent.id)
POST["form_type"] = str(Rating.TYPE_FILM)
POST["cancel_rating"] = "0"
rating_form = RatingForm(POST)
if rating_form.is_valid():
rating_helper.add_edit_rating(rating_form, self.u4)
else:
print "RATING FORM INVALID!"
result = recom_helper.get_recently_popular_films_query(self.u4.id)
self.assertEquals(len(result), 21)
def test_get_best_tci_users(self):
"""
Test for query that retrieves top users (based on common taste).
"""
do_create_comparators()
recom_helper = RecomHelper()
result = recom_helper.get_best_tci_users(self.u1)
self.assertEquals(len(result), 3)
def test_enrich_query_with_tags(self):
print "Test enrich query with tags"
recom_helper = RecomHelper()
film_helper = FilmHelper()
comedies = recom_helper.get_films_for_tag("comedy")
scifis = recom_helper.get_films_for_tag("sci-fi")
horrors = recom_helper.get_films_for_tag("horror")
thrillers = recom_helper.get_films_for_tag("thriller")
dramas = recom_helper.get_films_for_tag("drama")
fantasies = recom_helper.get_films_for_tag("fantasy")
mysteries = recom_helper.get_films_for_tag("mystery")
animations = recom_helper.get_films_for_tag("animation")
series = recom_helper.get_films_for_tag("series")
self.assertEquals(len(comedies), 5)
self.assertEquals(len(scifis), 6)
self.assertEquals(len(horrors), 3)
self.assertEquals(len(thrillers), 4)
self.assertEquals(len(dramas), 4)
self.assertEquals(len(fantasies), 3)
self.assertEquals(len(mysteries), 2)
self.assertEquals(len(animations), 3)
self.assertEquals(len(series), 4)
def test_get_all_ratings(self):
recom_helper = RecomHelper()
film_helper = FilmHelper()
film_with_director_and_actor = Film.objects.filter(title__startswith="Added film part 2")
related_directors = film_helper.get_film_directors(film_with_director_and_actor[0])
related_characters = film_helper.get_film_actors(film_with_director_and_actor[0])
related_actors = []
for character in related_characters:
actor = character.person
related_actors.append(actor)
non_rated_films = Film.objects.filter(rating__isnull=True, title__startswith="Added")
POST = {}
# POST['form_id'] = '1'
POST["rating"] = "6"
POST["object_id"] = str(non_rated_films[1].parent.id)
POST["form_type"] = str(Rating.TYPE_FILM)
POST["cancel_rating"] = "0"
rating_form = RatingForm(POST)
if rating_form.is_valid():
rating_helper.add_edit_rating(rating_form, self.u1)
else:
print "RATING FORM INVALID!"
all_ratings = recom_helper.get_all_ratings(self.u1.id)
self.assertEquals(len(all_ratings), 31)
director_ratings = recom_helper.get_all_ratings(self.u1.id, related_director=related_directors)
self.assertEquals(len(director_ratings), 1)
actor_ratings = recom_helper.get_all_ratings(self.u1.id, related_actor=related_actors)
self.assertEquals(len(actor_ratings), 1)
from_ratings = recom_helper.get_all_ratings(self.u1.id, year_from=1990)
self.assertEquals(len(from_ratings), 23)
to_ratings = recom_helper.get_all_ratings(self.u1.id, year_to=2000)
self.assertEquals(len(to_ratings), 22)
popularity_ratings = recom_helper.get_all_ratings(self.u1.id, popularity=5)
self.assertEquals(len(popularity_ratings), 8)
tag_ratings = recom_helper.get_all_ratings(self.u1.id, tags="comedy")
self.assertEquals(len(tag_ratings), 5)
def test_cache_count_ratings(self):
recom_helper = RecomHelper()
ratings_count = recom_helper.count_ratings(self.u3.id)
self.assertEquals(ratings_count, 15)
save_rating(self.f1, self.u3, 1)
ratings_count = recom_helper.count_ratings(self.u3.id)
self.assertEquals(ratings_count, 16)
def tearDown(self):
User.objects.all().delete()
Object.objects.all().delete()
Film.objects.all().delete()
FilmLog.objects.all().delete()
Rating.objects.all().delete()
Recommendation.objects.all().delete()
Person.objects.all().delete()
AddedCharacter.objects.all().delete()
AddedFilm.objects.all().delete()
def setUp(self):
User.objects.all().delete()
Object.objects.all().delete()
Film.objects.all().delete()
FilmLog.objects.all().delete()
Rating.objects.all().delete()
Recommendation.objects.all().delete()
Person.objects.all().delete()
AddedCharacter.objects.all().delete()
AddedFilm.objects.all().delete()
# set up users
self.u1 = User(username='******', email='*****@*****.**')
self.u1.save()
self.u2 = User.objects.create_user("root", "*****@*****.**", "root")
self.u2.user_permissions.add(
Permission.objects.get(codename="can_accept_added_films"))
self.u2.save()
self.u3 = User(username='******', email='*****@*****.**')
self.u3.save()
self.u4 = User(username='******', email='*****@*****.**')
self.u4.save()
self.u5 = User(username='******', email='*****@*****.**')
self.u5.save()
self.u6 = User(username='******', email='*****@*****.**')
self.u6.save()
# some persons
self.p1 = Person(name="Clint",
surname="Eastwood",
imdb_code=None,
type=Person.TYPE_PERSON)
self.p1.save()
self.p2 = Person(name="Sylvester",
surname="Stallone",
imdb_code=None,
type=Person.TYPE_PERSON)
self.p2.save()
self.p3 = Person(name="Jack",
surname="Mort",
imdb_code=None,
type=Person.TYPE_PERSON)
self.p3.save()
# set up films
self.f1 = Film(type=1,
permalink='przypadek',
imdb_code=111,
status=1,
version=1,
release_year=1999,
title='Przypadek',
popularity=20,
popularity_month=10)
self.f1.save()
self.f2 = Film(type=1,
permalink='wrestler',
imdb_code=112,
status=1,
version=1,
release_year=2008,
title='Wrestler',
popularity=10,
popularity_month=1)
self.f2.save()
self.f3 = Film(type=1,
permalink='american-history-x',
imdb_code=113,
status=1,
version=1,
release_year=1998,
title='American History X',
popularity=1,
popularity_month=1)
self.f3.save()
self.f4 = Film(type=1,
permalink='the-big-lebowski',
imdb_code=114,
status=1,
version=1,
release_year=1998,
title='The Big Lebowski',
popularity=1,
popularity_month=1)
self.f4.save()
self.f5 = Film(
type=1,
permalink='the-lord-of-the-rings-the-fellowship-of-the-ring',
imdb_code=115,
status=1,
version=1,
release_year=2001,
title='The Lord of the Rings: The Fellowship of the Ring',
popularity=1,
popularity_month=1)
self.f5.save()
self.f6 = Film(type=1,
permalink='raiders-of-the-lost-ark',
imdb_code=116,
status=1,
version=1,
release_year=1981,
title='Raiders of the Lost Ark',
popularity=1,
popularity_month=1)
self.f6.save()
self.f7 = Film(type=1,
permalink='the-alien',
imdb_code=117,
status=1,
version=1,
release_year=1979,
title='The Alien',
popularity=1,
popularity_month=1)
self.f7.save()
self.f8 = Film(type=1,
permalink='terminator-2-judgment-day',
imdb_code=118,
status=1,
version=1,
release_year=1991,
title='Terminator 2: Judgment Day',
popularity=1,
popularity_month=1)
self.f8.save()
self.f9 = Film(type=1,
permalink='przypadek',
imdb_code=119,
status=1,
version=1,
release_year=1999,
title='Przypadek',
popularity=20,
popularity_month=10)
self.f9.save()
self.f10 = Film(type=1,
permalink='wrestler',
imdb_code=120,
status=1,
version=1,
release_year=2008,
title='Wrestler',
popularity=10,
popularity_month=1)
self.f10.save()
self.f11 = Film(type=1,
permalink='american-history-x',
imdb_code=121,
status=1,
version=1,
release_year=1998,
title='American History X',
popularity=1,
popularity_month=1)
self.f11.save()
self.f12 = Film(type=1,
permalink='the-big-lebowski',
imdb_code=122,
status=1,
version=1,
release_year=1998,
title='The Big Lebowski',
popularity=1,
popularity_month=1)
self.f12.save()
self.f13 = Film(
type=1,
permalink='the-lord-of-the-rings-the-fellowship-of-the-ring',
imdb_code=123,
status=1,
version=1,
release_year=2001,
title='The Lord of the Rings: The Fellowship of the Ring',
popularity=1,
popularity_month=1)
self.f13.save()
self.f14 = Film(type=1,
permalink='raiders-of-the-lost-ark',
imdb_code=124,
status=1,
version=1,
release_year=1981,
title='Raiders of the Lost Ark',
popularity=1,
popularity_month=1)
self.f14.save()
self.f15 = Film(type=1,
permalink='the-alien',
imdb_code=125,
status=1,
version=1,
release_year=1979,
title='The Alien',
popularity=1,
popularity_month=1)
self.f15.save()
self.f16 = Film(type=1,
permalink='przypadek',
imdb_code=126,
status=1,
version=1,
release_year=1999,
title='Przypadek',
popularity=20,
popularity_month=10)
self.f16.save()
self.f17 = Film(type=1,
permalink='wrestler',
imdb_code=127,
status=1,
version=1,
release_year=2008,
title='Wrestler',
popularity=10,
popularity_month=1)
self.f17.save()
self.f18 = Film(type=1,
permalink='american-history-x',
imdb_code=128,
status=1,
version=1,
release_year=1998,
title='American History X',
popularity=1,
popularity_month=1)
self.f18.save()
self.f19 = Film(type=1,
permalink='the-big-lebowski',
imdb_code=129,
status=1,
version=1,
release_year=1998,
title='The Big Lebowski',
popularity=1,
popularity_month=1)
self.f19.save()
self.f20 = Film(
type=1,
permalink='the-lord-of-the-rings-the-fellowship-of-the-ring',
imdb_code=130,
status=1,
version=1,
release_year=2001,
title='The Lord of the Rings: The Fellowship of the Ring',
popularity=1,
popularity_month=1)
self.f20.save()
self.f21 = Film(type=1,
permalink='raiders-of-the-lost-ark',
imdb_code=131,
status=1,
version=1,
release_year=1981,
title='Raiders of the Lost Ark',
popularity=1,
popularity_month=1)
self.f21.save()
self.f22 = Film(type=1,
permalink='the-alien',
imdb_code=132,
status=1,
version=1,
release_year=1979,
title='The Alien',
popularity=1,
popularity_month=1)
self.f22.save()
self.f23 = Film(type=1,
permalink='terminator-2-judgment-day',
imdb_code=133,
status=1,
version=1,
release_year=1991,
title='Terminator 2: Judgment Day',
popularity=1,
popularity_month=1)
self.f23.save()
self.f24 = Film(type=1,
permalink='przypadek',
imdb_code=134,
status=1,
version=1,
release_year=1999,
title='Przypadek',
popularity=20,
popularity_month=10)
self.f24.save()
self.f25 = Film(type=1,
permalink='wrestler',
imdb_code=135,
status=1,
version=1,
release_year=2008,
title='Wrestler',
popularity=10,
popularity_month=1)
self.f25.save()
self.f26 = Film(type=1,
permalink='american-history-x',
imdb_code=136,
status=1,
version=1,
release_year=1998,
title='American History X',
popularity=1,
popularity_month=1)
self.f26.save()
self.f27 = Film(type=1,
permalink='the-big-lebowski',
imdb_code=137,
status=1,
version=1,
release_year=1998,
title='The Big Lebowski',
popularity=1,
popularity_month=1)
self.f27.save()
self.f28 = Film(
type=1,
permalink='the-lord-of-the-rings-the-fellowship-of-the-ring',
imdb_code=138,
status=1,
version=1,
release_year=2001,
title='The Lord of the Rings: The Fellowship of the Ring',
popularity=1,
popularity_month=1)
self.f28.save()
self.f29 = Film(type=1,
permalink='raiders-of-the-lost-ark',
imdb_code=139,
status=1,
version=1,
release_year=1981,
title='Raiders of the Lost Ark',
popularity=1,
popularity_month=1)
self.f29.save()
self.f30 = Film(type=1,
permalink='the-alien',
imdb_code=140,
status=1,
version=1,
release_year=1979,
title='The Alien',
popularity=1,
popularity_month=1)
self.f30.save()
self.f31 = AddedFilm(title='Added film part 1',
localized_title='Added film PL',
release_year=2010,
user=self.u1)
self.f31.save()
self.f31.directors.add(self.p1)
self.f31.save()
AddedCharacter.objects.create(added_film=self.f31,
person=self.p3,
character="Jack Mort")
self.f31.accept(self.u2)
self.f32 = AddedFilm(title='Added film part 2',
localized_title='Added film PL',
release_year=2010,
user=self.u1)
self.f32.save()
self.f32.directors.add(self.p2)
self.f32.save()
AddedCharacter.objects.create(added_film=self.f32,
person=self.p3,
character="Jack Mort")
self.f32.accept(self.u2)
#tagging
self.f1.save_tags('comedy drama horror animation mystery')
self.f2.save_tags('comedy sci-fi thriller series')
self.f3.save_tags('comedy')
self.f4.save_tags('sci-fi')
self.f5.save_tags('horror')
self.f6.save_tags('thriller')
self.f7.save_tags('drama')
self.f8.save_tags('series')
self.f9.save_tags('sci-fi')
self.f10.save_tags('fantasy')
self.f11.save_tags('mystery')
self.f12.save_tags('action')
self.f13.save_tags('adventure')
self.f14.save_tags('animation')
self.f15.save_tags('family')
self.f16.save_tags('thriller')
self.f17.save_tags('drama')
self.f18.save_tags('series')
self.f19.save_tags('sci-fi')
self.f20.save_tags('fantasy')
self.f21.save_tags('comedy')
self.f22.save_tags('animation')
self.f23.save_tags('comedy')
self.f24.save_tags('sci-fi')
self.f25.save_tags('horror')
self.f26.save_tags('thriller')
self.f27.save_tags('drama')
self.f28.save_tags('series')
self.f29.save_tags('sci-fi')
self.f30.save_tags('fantasy')
#First user ratings
save_rating(self.f1, self.u1, 1)
save_rating(self.f2, self.u1, 2)
save_rating(self.f3, self.u1, 3)
save_rating(self.f4, self.u1, 4)
save_rating(self.f5, self.u1, 5)
save_rating(self.f6, self.u1, 6)
save_rating(self.f7, self.u1, 7)
save_rating(self.f8, self.u1, 8)
save_rating(self.f9, self.u1, 9)
save_rating(self.f10, self.u1, 10)
save_rating(self.f11, self.u1, 1)
save_rating(self.f12, self.u1, 2)
save_rating(self.f13, self.u1, 3)
save_rating(self.f14, self.u1, 4)
save_rating(self.f15, self.u1, 5)
save_rating(self.f16, self.u1, 6)
save_rating(self.f17, self.u1, 7)
save_rating(self.f18, self.u1, 8)
save_rating(self.f19, self.u1, 9)
save_rating(self.f20, self.u1, 10)
save_rating(self.f21, self.u1, 1)
save_rating(self.f22, self.u1, 2)
save_rating(self.f23, self.u1, 3)
save_rating(self.f24, self.u1, 4)
save_rating(self.f25, self.u1, 5)
save_rating(self.f26, self.u1, 6)
save_rating(self.f27, self.u1, 7)
save_rating(self.f28, self.u1, 8)
save_rating(self.f29, self.u1, 9)
save_rating(self.f30, self.u1, 10)
#Second user ratings - more than minimal common films requirement (15)
save_rating(self.f1, self.u2, 1)
save_rating(self.f2, self.u2, 2)
save_rating(self.f3, self.u2, 3)
save_rating(self.f4, self.u2, 4)
save_rating(self.f5, self.u2, 5)
save_rating(self.f6, self.u2, 6)
save_rating(self.f7, self.u2, 7)
save_rating(self.f8, self.u2, 8)
save_rating(self.f9, self.u2, 9)
save_rating(self.f10, self.u2, 10)
save_rating(self.f11, self.u2, 1)
#save_rating(self.f12, self.u2, 2)
save_rating(self.f13, self.u2, 3)
#save_rating(self.f14, self.u2, 4)
save_rating(self.f15, self.u2, 5)
#save_rating(self.f16, self.u2, 6)
save_rating(self.f17, self.u2, 7)
#save_rating(self.f18, self.u2, 8)
save_rating(self.f19, self.u2, 9)
#save_rating(self.f20, self.u2, 10)
save_rating(self.f21, self.u2, 1)
#save_rating(self.f22, self.u2, 2)
save_rating(self.f23, self.u2, 3)
#save_rating(self.f24, self.u2, 4)
save_rating(self.f25, self.u2, 5)
#save_rating(self.f26, self.u2, 6)
save_rating(self.f27, self.u2, 7)
#save_rating(self.f28, self.u2, 8)
save_rating(self.f29, self.u2, 9)
#save_rating(self.f30, self.u2, 10)
# Third user ratings - equal with minimal common films requirement (15)
#save_rating(self.f1, self.u3, 1)
#save_rating(self.f2, self.u3, 2)
#save_rating(self.f3, self.u3, 3)
#save_rating(self.f4, self.u3, 4)
#save_rating(self.f5, self.u3, 5)
#save_rating(self.f6, self.u3, 6)
#save_rating(self.f7, self.u3, 7)
#save_rating(self.f8, self.u3, 8)
#save_rating(self.f9, self.u3, 9)
#save_rating(self.f10, self.u3, 10)
#save_rating(self.f11, self.u3, 1)
#save_rating(self.f12, self.u3, 2)
#save_rating(self.f13, self.u3, 3)
#save_rating(self.f14, self.u3, 4)
#save_rating(self.f15, self.u3, 5)
save_rating(self.f16, self.u3, 6)
save_rating(self.f17, self.u3, 7)
save_rating(self.f18, self.u3, 8)
save_rating(self.f19, self.u3, 9)
save_rating(self.f20, self.u3, 10)
save_rating(self.f21, self.u3, 1)
save_rating(self.f22, self.u3, 2)
save_rating(self.f23, self.u3, 3)
save_rating(self.f24, self.u3, 4)
save_rating(self.f25, self.u3, 5)
save_rating(self.f26, self.u3, 6)
save_rating(self.f27, self.u3, 7)
save_rating(self.f28, self.u3, 8)
save_rating(self.f29, self.u3, 9)
save_rating(self.f30, self.u3, 10)
# Fourth user ratings - less than minimal films requirement
#save_rating(self.f1, self.u4, 1)
#save_rating(self.f2, self.u4, 2)
#save_rating(self.f3, self.u4, 3)
#save_rating(self.f4, self.u4, 4)
#save_rating(self.f5, self.u4, 5)
#save_rating(self.f6, self.u4, 6)
#save_rating(self.f7, self.u4, 7)
#save_rating(self.f8, self.u4, 8)
#save_rating(self.f9, self.u4, 9)
#save_rating(self.f10, self.u4, 10)
#save_rating(self.f11, self.u4, 1)
#save_rating(self.f12, self.u4, 2)
#save_rating(self.f13, self.u4, 3)
#save_rating(self.f14, self.u4, 4)
#save_rating(self.f15, self.u4, 5)
#save_rating(self.f16, self.u4, 6)
#save_rating(self.f17, self.u4, 7)
#save_rating(self.f18, self.u4, 8)
#save_rating(self.f19, self.u4, 9)
#save_rating(self.f20, self.u4, 10)
save_rating(self.f21, self.u4, 1)
save_rating(self.f22, self.u4, 2)
save_rating(self.f23, self.u4, 3)
save_rating(self.f24, self.u4, 4)
save_rating(self.f25, self.u4, 5)
save_rating(self.f26, self.u4, 6)
save_rating(self.f27, self.u4, 7)
save_rating(self.f28, self.u4, 8)
save_rating(self.f29, self.u4, 9)
save_rating(self.f30, self.u4, 10)
# Fifth user ratings - the same films rated (1st and 5th user) but different
# ratings
#save_rating(self.f1, self.u5, 10)
#save_rating(self.f2, self.u5, 9)
#save_rating(self.f3, self.u5, 8)
#save_rating(self.f4, self.u5, 7)
#save_rating(self.f5, self.u5, 6)
#save_rating(self.f6, self.u5, 5)
#save_rating(self.f7, self.u5, 4)
#save_rating(self.f8, self.u5, 3)
#save_rating(self.f9, self.u5, 2)
#save_rating(self.f10, self.u5, 1)
#save_rating(self.f11, self.u5, 10)
#save_rating(self.f12, self.u5, 9)
#save_rating(self.f13, self.u5, 8)
#save_rating(self.f14, self.u5, 7)
#save_rating(self.f15, self.u5, 6)
#save_rating(self.f16, self.u5, 5)
#save_rating(self.f17, self.u5, 4)
#save_rating(self.f18, self.u5, 3)
#save_rating(self.f19, self.u5, 2)
#save_rating(self.f20, self.u5, 1)
#save_rating(self.f21, self.u5, 10)
#save_rating(self.f22, self.u5, 9)
#save_rating(self.f23, self.u5, 8)
#save_rating(self.f24, self.u5, 7)
#save_rating(self.f25, self.u5, 6)
#save_rating(self.f26, self.u5, 5)
#save_rating(self.f27, self.u5, 4)
#save_rating(self.f28, self.u5, 3)
#save_rating(self.f29, self.u5, 2)
#save_rating(self.f30, self.u5, 1)
# Sixth user ratings - the same films rated (1st and 5th user) but different
# ratings
save_rating(self.f1, self.u6, 10)
save_rating(self.f2, self.u6, 10)
save_rating(self.f3, self.u6, 10)
save_rating(self.f4, self.u6, 10)
save_rating(self.f5, self.u6, 10)
save_rating(self.f6, self.u6, 10)
save_rating(self.f7, self.u6, 10)
save_rating(self.f8, self.u6, 10)
save_rating(self.f9, self.u6, 10)
save_rating(self.f10, self.u6, 10)
save_rating(self.f11, self.u6, 10)
save_rating(self.f12, self.u6, 10)
save_rating(self.f13, self.u6, 10)
save_rating(self.f14, self.u6, 10)
save_rating(self.f15, self.u6, 10)
save_rating(self.f16, self.u6, 10)
save_rating(self.f17, self.u6, 10)
save_rating(self.f18, self.u6, 10)
save_rating(self.f19, self.u6, 10)
save_rating(self.f20, self.u6, 10)
save_rating(self.f21, self.u6, 10)
save_rating(self.f22, self.u6, 10)
save_rating(self.f23, self.u6, 10)
save_rating(self.f24, self.u6, 10)
save_rating(self.f25, self.u6, 10)
save_rating(self.f26, self.u6, 10)
save_rating(self.f27, self.u6, 10)
save_rating(self.f28, self.u6, 10)
save_rating(self.f29, self.u6, 10)
save_rating(self.f30, self.u6, 10)
class QueriesTestCase(TestCase):
def setUp(self):
User.objects.all().delete()
Object.objects.all().delete()
Film.objects.all().delete()
FilmLog.objects.all().delete()
Rating.objects.all().delete()
Recommendation.objects.all().delete()
Person.objects.all().delete()
AddedCharacter.objects.all().delete()
AddedFilm.objects.all().delete()
# set up users
self.u1 = User(username='******', email='*****@*****.**')
self.u1.save()
self.u2 = User.objects.create_user("root", "*****@*****.**", "root")
self.u2.user_permissions.add(
Permission.objects.get(codename="can_accept_added_films"))
self.u2.save()
self.u3 = User(username='******', email='*****@*****.**')
self.u3.save()
self.u4 = User(username='******', email='*****@*****.**')
self.u4.save()
self.u5 = User(username='******', email='*****@*****.**')
self.u5.save()
self.u6 = User(username='******', email='*****@*****.**')
self.u6.save()
# some persons
self.p1 = Person(name="Clint",
surname="Eastwood",
imdb_code=None,
type=Person.TYPE_PERSON)
self.p1.save()
self.p2 = Person(name="Sylvester",
surname="Stallone",
imdb_code=None,
type=Person.TYPE_PERSON)
self.p2.save()
self.p3 = Person(name="Jack",
surname="Mort",
imdb_code=None,
type=Person.TYPE_PERSON)
self.p3.save()
# set up films
self.f1 = Film(type=1,
permalink='przypadek',
imdb_code=111,
status=1,
version=1,
release_year=1999,
title='Przypadek',
popularity=20,
popularity_month=10)
self.f1.save()
self.f2 = Film(type=1,
permalink='wrestler',
imdb_code=112,
status=1,
version=1,
release_year=2008,
title='Wrestler',
popularity=10,
popularity_month=1)
self.f2.save()
self.f3 = Film(type=1,
permalink='american-history-x',
imdb_code=113,
status=1,
version=1,
release_year=1998,
title='American History X',
popularity=1,
popularity_month=1)
self.f3.save()
self.f4 = Film(type=1,
permalink='the-big-lebowski',
imdb_code=114,
status=1,
version=1,
release_year=1998,
title='The Big Lebowski',
popularity=1,
popularity_month=1)
self.f4.save()
self.f5 = Film(
type=1,
permalink='the-lord-of-the-rings-the-fellowship-of-the-ring',
imdb_code=115,
status=1,
version=1,
release_year=2001,
title='The Lord of the Rings: The Fellowship of the Ring',
popularity=1,
popularity_month=1)
self.f5.save()
self.f6 = Film(type=1,
permalink='raiders-of-the-lost-ark',
imdb_code=116,
status=1,
version=1,
release_year=1981,
title='Raiders of the Lost Ark',
popularity=1,
popularity_month=1)
self.f6.save()
self.f7 = Film(type=1,
permalink='the-alien',
imdb_code=117,
status=1,
version=1,
release_year=1979,
title='The Alien',
popularity=1,
popularity_month=1)
self.f7.save()
self.f8 = Film(type=1,
permalink='terminator-2-judgment-day',
imdb_code=118,
status=1,
version=1,
release_year=1991,
title='Terminator 2: Judgment Day',
popularity=1,
popularity_month=1)
self.f8.save()
self.f9 = Film(type=1,
permalink='przypadek',
imdb_code=119,
status=1,
version=1,
release_year=1999,
title='Przypadek',
popularity=20,
popularity_month=10)
self.f9.save()
self.f10 = Film(type=1,
permalink='wrestler',
imdb_code=120,
status=1,
version=1,
release_year=2008,
title='Wrestler',
popularity=10,
popularity_month=1)
self.f10.save()
self.f11 = Film(type=1,
permalink='american-history-x',
imdb_code=121,
status=1,
version=1,
release_year=1998,
title='American History X',
popularity=1,
popularity_month=1)
self.f11.save()
self.f12 = Film(type=1,
permalink='the-big-lebowski',
imdb_code=122,
status=1,
version=1,
release_year=1998,
title='The Big Lebowski',
popularity=1,
popularity_month=1)
self.f12.save()
self.f13 = Film(
type=1,
permalink='the-lord-of-the-rings-the-fellowship-of-the-ring',
imdb_code=123,
status=1,
version=1,
release_year=2001,
title='The Lord of the Rings: The Fellowship of the Ring',
popularity=1,
popularity_month=1)
self.f13.save()
self.f14 = Film(type=1,
permalink='raiders-of-the-lost-ark',
imdb_code=124,
status=1,
version=1,
release_year=1981,
title='Raiders of the Lost Ark',
popularity=1,
popularity_month=1)
self.f14.save()
self.f15 = Film(type=1,
permalink='the-alien',
imdb_code=125,
status=1,
version=1,
release_year=1979,
title='The Alien',
popularity=1,
popularity_month=1)
self.f15.save()
self.f16 = Film(type=1,
permalink='przypadek',
imdb_code=126,
status=1,
version=1,
release_year=1999,
title='Przypadek',
popularity=20,
popularity_month=10)
self.f16.save()
self.f17 = Film(type=1,
permalink='wrestler',
imdb_code=127,
status=1,
version=1,
release_year=2008,
title='Wrestler',
popularity=10,
popularity_month=1)
self.f17.save()
self.f18 = Film(type=1,
permalink='american-history-x',
imdb_code=128,
status=1,
version=1,
release_year=1998,
title='American History X',
popularity=1,
popularity_month=1)
self.f18.save()
self.f19 = Film(type=1,
permalink='the-big-lebowski',
imdb_code=129,
status=1,
version=1,
release_year=1998,
title='The Big Lebowski',
popularity=1,
popularity_month=1)
self.f19.save()
self.f20 = Film(
type=1,
permalink='the-lord-of-the-rings-the-fellowship-of-the-ring',
imdb_code=130,
status=1,
version=1,
release_year=2001,
title='The Lord of the Rings: The Fellowship of the Ring',
popularity=1,
popularity_month=1)
self.f20.save()
self.f21 = Film(type=1,
permalink='raiders-of-the-lost-ark',
imdb_code=131,
status=1,
version=1,
release_year=1981,
title='Raiders of the Lost Ark',
popularity=1,
popularity_month=1)
self.f21.save()
self.f22 = Film(type=1,
permalink='the-alien',
imdb_code=132,
status=1,
version=1,
release_year=1979,
title='The Alien',
popularity=1,
popularity_month=1)
self.f22.save()
self.f23 = Film(type=1,
permalink='terminator-2-judgment-day',
imdb_code=133,
status=1,
version=1,
release_year=1991,
title='Terminator 2: Judgment Day',
popularity=1,
popularity_month=1)
self.f23.save()
self.f24 = Film(type=1,
permalink='przypadek',
imdb_code=134,
status=1,
version=1,
release_year=1999,
title='Przypadek',
popularity=20,
popularity_month=10)
self.f24.save()
self.f25 = Film(type=1,
permalink='wrestler',
imdb_code=135,
status=1,
version=1,
release_year=2008,
title='Wrestler',
popularity=10,
popularity_month=1)
self.f25.save()
self.f26 = Film(type=1,
permalink='american-history-x',
imdb_code=136,
status=1,
version=1,
release_year=1998,
title='American History X',
popularity=1,
popularity_month=1)
self.f26.save()
self.f27 = Film(type=1,
permalink='the-big-lebowski',
imdb_code=137,
status=1,
version=1,
release_year=1998,
title='The Big Lebowski',
popularity=1,
popularity_month=1)
self.f27.save()
self.f28 = Film(
type=1,
permalink='the-lord-of-the-rings-the-fellowship-of-the-ring',
imdb_code=138,
status=1,
version=1,
release_year=2001,
title='The Lord of the Rings: The Fellowship of the Ring',
popularity=1,
popularity_month=1)
self.f28.save()
self.f29 = Film(type=1,
permalink='raiders-of-the-lost-ark',
imdb_code=139,
status=1,
version=1,
release_year=1981,
title='Raiders of the Lost Ark',
popularity=1,
popularity_month=1)
self.f29.save()
self.f30 = Film(type=1,
permalink='the-alien',
imdb_code=140,
status=1,
version=1,
release_year=1979,
title='The Alien',
popularity=1,
popularity_month=1)
self.f30.save()
self.f31 = AddedFilm(title='Added film part 1',
localized_title='Added film PL',
release_year=2010,
user=self.u1)
self.f31.save()
self.f31.directors.add(self.p1)
self.f31.save()
AddedCharacter.objects.create(added_film=self.f31,
person=self.p3,
character="Jack Mort")
self.f31.accept(self.u2)
self.f32 = AddedFilm(title='Added film part 2',
localized_title='Added film PL',
release_year=2010,
user=self.u1)
self.f32.save()
self.f32.directors.add(self.p2)
self.f32.save()
AddedCharacter.objects.create(added_film=self.f32,
person=self.p3,
character="Jack Mort")
self.f32.accept(self.u2)
#tagging
self.f1.save_tags('comedy drama horror animation mystery')
self.f2.save_tags('comedy sci-fi thriller series')
self.f3.save_tags('comedy')
self.f4.save_tags('sci-fi')
self.f5.save_tags('horror')
self.f6.save_tags('thriller')
self.f7.save_tags('drama')
self.f8.save_tags('series')
self.f9.save_tags('sci-fi')
self.f10.save_tags('fantasy')
self.f11.save_tags('mystery')
self.f12.save_tags('action')
self.f13.save_tags('adventure')
self.f14.save_tags('animation')
self.f15.save_tags('family')
self.f16.save_tags('thriller')
self.f17.save_tags('drama')
self.f18.save_tags('series')
self.f19.save_tags('sci-fi')
self.f20.save_tags('fantasy')
self.f21.save_tags('comedy')
self.f22.save_tags('animation')
self.f23.save_tags('comedy')
self.f24.save_tags('sci-fi')
self.f25.save_tags('horror')
self.f26.save_tags('thriller')
self.f27.save_tags('drama')
self.f28.save_tags('series')
self.f29.save_tags('sci-fi')
self.f30.save_tags('fantasy')
#First user ratings
save_rating(self.f1, self.u1, 1)
save_rating(self.f2, self.u1, 2)
save_rating(self.f3, self.u1, 3)
save_rating(self.f4, self.u1, 4)
save_rating(self.f5, self.u1, 5)
save_rating(self.f6, self.u1, 6)
save_rating(self.f7, self.u1, 7)
save_rating(self.f8, self.u1, 8)
save_rating(self.f9, self.u1, 9)
save_rating(self.f10, self.u1, 10)
save_rating(self.f11, self.u1, 1)
save_rating(self.f12, self.u1, 2)
save_rating(self.f13, self.u1, 3)
save_rating(self.f14, self.u1, 4)
save_rating(self.f15, self.u1, 5)
save_rating(self.f16, self.u1, 6)
save_rating(self.f17, self.u1, 7)
save_rating(self.f18, self.u1, 8)
save_rating(self.f19, self.u1, 9)
save_rating(self.f20, self.u1, 10)
save_rating(self.f21, self.u1, 1)
save_rating(self.f22, self.u1, 2)
save_rating(self.f23, self.u1, 3)
save_rating(self.f24, self.u1, 4)
save_rating(self.f25, self.u1, 5)
save_rating(self.f26, self.u1, 6)
save_rating(self.f27, self.u1, 7)
save_rating(self.f28, self.u1, 8)
save_rating(self.f29, self.u1, 9)
save_rating(self.f30, self.u1, 10)
#Second user ratings - more than minimal common films requirement (15)
save_rating(self.f1, self.u2, 1)
save_rating(self.f2, self.u2, 2)
save_rating(self.f3, self.u2, 3)
save_rating(self.f4, self.u2, 4)
save_rating(self.f5, self.u2, 5)
save_rating(self.f6, self.u2, 6)
save_rating(self.f7, self.u2, 7)
save_rating(self.f8, self.u2, 8)
save_rating(self.f9, self.u2, 9)
save_rating(self.f10, self.u2, 10)
save_rating(self.f11, self.u2, 1)
#save_rating(self.f12, self.u2, 2)
save_rating(self.f13, self.u2, 3)
#save_rating(self.f14, self.u2, 4)
save_rating(self.f15, self.u2, 5)
#save_rating(self.f16, self.u2, 6)
save_rating(self.f17, self.u2, 7)
#save_rating(self.f18, self.u2, 8)
save_rating(self.f19, self.u2, 9)
#save_rating(self.f20, self.u2, 10)
save_rating(self.f21, self.u2, 1)
#save_rating(self.f22, self.u2, 2)
save_rating(self.f23, self.u2, 3)
#save_rating(self.f24, self.u2, 4)
save_rating(self.f25, self.u2, 5)
#save_rating(self.f26, self.u2, 6)
save_rating(self.f27, self.u2, 7)
#save_rating(self.f28, self.u2, 8)
save_rating(self.f29, self.u2, 9)
#save_rating(self.f30, self.u2, 10)
# Third user ratings - equal with minimal common films requirement (15)
#save_rating(self.f1, self.u3, 1)
#save_rating(self.f2, self.u3, 2)
#save_rating(self.f3, self.u3, 3)
#save_rating(self.f4, self.u3, 4)
#save_rating(self.f5, self.u3, 5)
#save_rating(self.f6, self.u3, 6)
#save_rating(self.f7, self.u3, 7)
#save_rating(self.f8, self.u3, 8)
#save_rating(self.f9, self.u3, 9)
#save_rating(self.f10, self.u3, 10)
#save_rating(self.f11, self.u3, 1)
#save_rating(self.f12, self.u3, 2)
#save_rating(self.f13, self.u3, 3)
#save_rating(self.f14, self.u3, 4)
#save_rating(self.f15, self.u3, 5)
save_rating(self.f16, self.u3, 6)
save_rating(self.f17, self.u3, 7)
save_rating(self.f18, self.u3, 8)
save_rating(self.f19, self.u3, 9)
save_rating(self.f20, self.u3, 10)
save_rating(self.f21, self.u3, 1)
save_rating(self.f22, self.u3, 2)
save_rating(self.f23, self.u3, 3)
save_rating(self.f24, self.u3, 4)
save_rating(self.f25, self.u3, 5)
save_rating(self.f26, self.u3, 6)
save_rating(self.f27, self.u3, 7)
save_rating(self.f28, self.u3, 8)
save_rating(self.f29, self.u3, 9)
save_rating(self.f30, self.u3, 10)
# Fourth user ratings - less than minimal films requirement
#save_rating(self.f1, self.u4, 1)
#save_rating(self.f2, self.u4, 2)
#save_rating(self.f3, self.u4, 3)
#save_rating(self.f4, self.u4, 4)
#save_rating(self.f5, self.u4, 5)
#save_rating(self.f6, self.u4, 6)
#save_rating(self.f7, self.u4, 7)
#save_rating(self.f8, self.u4, 8)
#save_rating(self.f9, self.u4, 9)
#save_rating(self.f10, self.u4, 10)
#save_rating(self.f11, self.u4, 1)
#save_rating(self.f12, self.u4, 2)
#save_rating(self.f13, self.u4, 3)
#save_rating(self.f14, self.u4, 4)
#save_rating(self.f15, self.u4, 5)
#save_rating(self.f16, self.u4, 6)
#save_rating(self.f17, self.u4, 7)
#save_rating(self.f18, self.u4, 8)
#save_rating(self.f19, self.u4, 9)
#save_rating(self.f20, self.u4, 10)
save_rating(self.f21, self.u4, 1)
save_rating(self.f22, self.u4, 2)
save_rating(self.f23, self.u4, 3)
save_rating(self.f24, self.u4, 4)
save_rating(self.f25, self.u4, 5)
save_rating(self.f26, self.u4, 6)
save_rating(self.f27, self.u4, 7)
save_rating(self.f28, self.u4, 8)
save_rating(self.f29, self.u4, 9)
save_rating(self.f30, self.u4, 10)
# Fifth user ratings - the same films rated (1st and 5th user) but different
# ratings
#save_rating(self.f1, self.u5, 10)
#save_rating(self.f2, self.u5, 9)
#save_rating(self.f3, self.u5, 8)
#save_rating(self.f4, self.u5, 7)
#save_rating(self.f5, self.u5, 6)
#save_rating(self.f6, self.u5, 5)
#save_rating(self.f7, self.u5, 4)
#save_rating(self.f8, self.u5, 3)
#save_rating(self.f9, self.u5, 2)
#save_rating(self.f10, self.u5, 1)
#save_rating(self.f11, self.u5, 10)
#save_rating(self.f12, self.u5, 9)
#save_rating(self.f13, self.u5, 8)
#save_rating(self.f14, self.u5, 7)
#save_rating(self.f15, self.u5, 6)
#save_rating(self.f16, self.u5, 5)
#save_rating(self.f17, self.u5, 4)
#save_rating(self.f18, self.u5, 3)
#save_rating(self.f19, self.u5, 2)
#save_rating(self.f20, self.u5, 1)
#save_rating(self.f21, self.u5, 10)
#save_rating(self.f22, self.u5, 9)
#save_rating(self.f23, self.u5, 8)
#save_rating(self.f24, self.u5, 7)
#save_rating(self.f25, self.u5, 6)
#save_rating(self.f26, self.u5, 5)
#save_rating(self.f27, self.u5, 4)
#save_rating(self.f28, self.u5, 3)
#save_rating(self.f29, self.u5, 2)
#save_rating(self.f30, self.u5, 1)
# Sixth user ratings - the same films rated (1st and 5th user) but different
# ratings
save_rating(self.f1, self.u6, 10)
save_rating(self.f2, self.u6, 10)
save_rating(self.f3, self.u6, 10)
save_rating(self.f4, self.u6, 10)
save_rating(self.f5, self.u6, 10)
save_rating(self.f6, self.u6, 10)
save_rating(self.f7, self.u6, 10)
save_rating(self.f8, self.u6, 10)
save_rating(self.f9, self.u6, 10)
save_rating(self.f10, self.u6, 10)
save_rating(self.f11, self.u6, 10)
save_rating(self.f12, self.u6, 10)
save_rating(self.f13, self.u6, 10)
save_rating(self.f14, self.u6, 10)
save_rating(self.f15, self.u6, 10)
save_rating(self.f16, self.u6, 10)
save_rating(self.f17, self.u6, 10)
save_rating(self.f18, self.u6, 10)
save_rating(self.f19, self.u6, 10)
save_rating(self.f20, self.u6, 10)
save_rating(self.f21, self.u6, 10)
save_rating(self.f22, self.u6, 10)
save_rating(self.f23, self.u6, 10)
save_rating(self.f24, self.u6, 10)
save_rating(self.f25, self.u6, 10)
save_rating(self.f26, self.u6, 10)
save_rating(self.f27, self.u6, 10)
save_rating(self.f28, self.u6, 10)
save_rating(self.f29, self.u6, 10)
save_rating(self.f30, self.u6, 10)
@unittest.skipIf(not is_postgres, "sqlite is not supported yet")
def test_get_recently_popular_films_query(self):
print "Test get recently popular films query"
"""
Test for query which gets the most popular films for the rating machine.
First it should return all films, then we rate one and in the second call
it should only retrieve the other film.
"""
recom_helper = RecomHelper()
film_helper = FilmHelper()
u1_result = recom_helper.get_recently_popular_films_query(self.u1.id)
self.assertEquals(len(u1_result), 2)
u2_result = recom_helper.get_recently_popular_films_query(self.u2.id)
self.assertEquals(len(u2_result), 12)
film_with_director_and_actor = Film.objects.filter(
title__startswith="Added film part 1")
related_directors = film_helper.get_film_directors(
film_with_director_and_actor[0])
related_characters = film_helper.get_film_actors(
film_with_director_and_actor[0])
related_actors = []
for character in related_characters:
actor = character.person
related_actors.append(actor)
POST = {}
# TODO check why it starts breaking (form invalid) when form_id is set
#POST['form_id'] = '1'
POST['rating'] = '6'
POST['object_id'] = str(self.f12.parent.id)
POST['form_type'] = str(Rating.TYPE_FILM)
POST['cancel_rating'] = '0'
rating_form = RatingForm(POST)
if rating_form.is_valid():
rating_helper.add_edit_rating(rating_form, self.u2)
else:
print "RATING FORM INVALID!"
result = recom_helper.get_recently_popular_films_query(self.u2.id)
self.assertEquals(len(result), 11)
# test filtering
result = recom_helper.get_recently_popular_films_query(self.u5.id)
self.assertEquals(len(result), 32)
result = recom_helper.get_recently_popular_films_query(
self.u5.id, year_from='1990')
self.assertEquals(len(result), 24)
result = recom_helper.get_recently_popular_films_query(self.u5.id,
year_to='2000')
self.assertEquals(len(result), 22)
result = recom_helper.get_recently_popular_films_query(
self.u5.id, year_from='1990', year_to='2000')
self.assertEquals(len(result), 14)
result = recom_helper.get_recently_popular_films_query(self.u5.id,
popularity='10')
self.assertEquals(len(result), 8)
result = recom_helper.get_recently_popular_films_query(
self.u5.id, year_from='1990', year_to='2000', popularity='10')
self.assertEquals(len(result), 4)
rated_films = [
self.f1.id, self.f3.id, self.f5.id, self.f7.id, self.f9.id
]
result = recom_helper.get_recently_popular_films_query(
self.u5.id, rated_films=rated_films)
self.assertEquals(len(result), 27)
result = recom_helper.get_recently_popular_films_query(
self.u5.id,
year_from='1990',
year_to='2000',
rated_films=rated_films,
popularity='10')
self.assertEquals(len(result), 2)
result = recom_helper.get_recently_popular_films_query(
self.u5.id, related_director=related_directors)
self.assertEquals(len(result), 1)
result = recom_helper.get_recently_popular_films_query(
self.u5.id, related_actor=related_actors)
self.assertEquals(len(result), 2)
result = recom_helper.get_recently_popular_films_query(
self.u5.id,
related_director=related_directors,
related_actor=related_actors)
self.assertEquals(len(result), 1)
result = recom_helper.get_recently_popular_films_query(
self.u5.id,
year_from='1990',
year_to='2011',
related_director=related_directors,
related_actor=related_actors)
self.assertEquals(len(result), 1)
# @unittest.skipIf(not is_postgres, "postgres dependend")
# def test_get_best_psi_films_queryset_alg2(self):
# """
# Test for query that retrieves top recommendations.
# First it should return all films, then we rate one and in the second call
# it should only retrieve the other film.
# """
#
# recom_helper = RecomHelper()
# film_helper = FilmHelper()
#
# # no recommendations yet, the result should be 0
# result = recom_helper.get_best_psi_films_queryset_alg2(self.u4.id)
# self.assertEquals(len(result), 0)
#
# # simulate generating recommendations (for all films) and test again, it should retrieve all films
# r1 = Recommendation()
# r1.guess_rating_alg2 = 7
# r1.film = self.f19
# r1.user = self.u4
# r1.save()
# r2 = Recommendation()
# r2.guess_rating_alg2 = 8
# r2.film = self.f20
# r2.user = self.u4
# r2.save()
# result = recom_helper.get_best_psi_films_queryset_alg2(self.u4.id)
# self.assertEquals(len(result), 2)
#
# # rate one film, it should be removed from the list
# POST = {}
# # TODO check why it starts breaking (form invalid) when form_id is set
# #POST['form_id'] = '1'
# POST['rating'] = '6'
# POST['object_id'] = str(self.f20.parent.id)
# POST['form_type'] = str(Rating.TYPE_FILM)
# POST['cancel_rating'] = '0'
# rating_form = RatingForm(POST)
# if rating_form.is_valid():
# rating_helper.add_edit_rating(rating_form, self.u4)
# else:
# print "RATING FORM INVALID!"
#
# result = recom_helper.get_recently_popular_films_query(self.u4.id)
# self.assertEquals(len(result), 21)
@unittest.skip("temporary disabled")
def test_get_best_tci_users(self):
"""
Test for query that retrieves top users (based on common taste).
"""
do_create_comparators()
recom_helper = RecomHelper()
result = recom_helper.get_best_tci_users(self.u1)
print "result -->", result
self.assertEquals(len(result), 4)
def test_enrich_query_with_tags(self):
print "Test enrich query with tags"
recom_helper = RecomHelper()
film_helper = FilmHelper()
comedies = recom_helper.get_films_for_tag('comedy')
scifis = recom_helper.get_films_for_tag('sci-fi')
horrors = recom_helper.get_films_for_tag('horror')
thrillers = recom_helper.get_films_for_tag('thriller')
dramas = recom_helper.get_films_for_tag('drama')
fantasies = recom_helper.get_films_for_tag('fantasy')
mysteries = recom_helper.get_films_for_tag('mystery')
animations = recom_helper.get_films_for_tag('animation')
series = recom_helper.get_films_for_tag('series')
self.assertEquals(len(comedies), 5)
self.assertEquals(len(scifis), 6)
self.assertEquals(len(horrors), 3)
self.assertEquals(len(thrillers), 4)
self.assertEquals(len(dramas), 4)
self.assertEquals(len(fantasies), 3)
self.assertEquals(len(mysteries), 2)
self.assertEquals(len(animations), 3)
self.assertEquals(len(series), 4)
def test_get_all_ratings(self):
recom_helper = RecomHelper()
film_helper = FilmHelper()
film_with_director_and_actor = Film.objects.filter(
title__startswith="Added film part 2")
related_directors = film_helper.get_film_directors(
film_with_director_and_actor[0])
related_characters = film_helper.get_film_actors(
film_with_director_and_actor[0])
related_actors = []
for character in related_characters:
actor = character.person
related_actors.append(actor)
non_rated_films = Film.objects.filter(rating__isnull=True,
title__startswith="Added")
POST = {}
#POST['form_id'] = '1'
POST['rating'] = '6'
POST['object_id'] = str(non_rated_films[1].parent.id)
POST['form_type'] = str(Rating.TYPE_FILM)
POST['cancel_rating'] = '0'
rating_form = RatingForm(POST)
if rating_form.is_valid():
rating_helper.add_edit_rating(rating_form, self.u1)
else:
print "RATING FORM INVALID!"
all_ratings = recom_helper.get_all_ratings(self.u1.id)
self.assertEquals(len(all_ratings), 31)
director_ratings = recom_helper.get_all_ratings(
self.u1.id, related_director=related_directors)
self.assertEquals(len(director_ratings), 1)
actor_ratings = recom_helper.get_all_ratings(
self.u1.id, related_actor=related_actors)
self.assertEquals(len(actor_ratings), 1)
from_ratings = recom_helper.get_all_ratings(self.u1.id, year_from=1990)
self.assertEquals(len(from_ratings), 23)
to_ratings = recom_helper.get_all_ratings(self.u1.id, year_to=2000)
self.assertEquals(len(to_ratings), 22)
popularity_ratings = recom_helper.get_all_ratings(self.u1.id,
popularity=5)
self.assertEquals(len(popularity_ratings), 8)
tag_ratings = recom_helper.get_all_ratings(self.u1.id, tags='comedy')
self.assertEquals(len(tag_ratings), 5)
def test_cache_count_ratings(self):
recom_helper = RecomHelper()
ratings_count = recom_helper.count_ratings(self.u3.id)
self.assertEquals(ratings_count, 15)
save_rating(self.f1, self.u3, 1)
ratings_count = recom_helper.count_ratings(self.u3.id)
self.assertEquals(ratings_count, 16)
def tearDown(self):
User.objects.all().delete()
Object.objects.all().delete()
Film.objects.all().delete()
FilmLog.objects.all().delete()
Rating.objects.all().delete()
Recommendation.objects.all().delete()
Person.objects.all().delete()
AddedCharacter.objects.all().delete()
AddedFilm.objects.all().delete()
