def test_filter_by_date(self):
result = lib.procesar_queries([
"filter_by_date",
["popular_movies", ["load_database"], 50, 100, "RT"], 2009
])
movies = {int(movie.id) for movie in result}
self.assertEqual(movies, self.results["filter_by_date"])
def test_best_comments(self):
preprocesamiento.reviews_writer("clean_reviews.csv",
"TestingDatabase/reviews.csv")
result = lib.procesar_queries([
"best_comments",
["take_movie_while", ["load_database"], "RT", ">", 30], 1
])
movies = {int(movie.id) for movie in result}
self.assertEqual(movies, self.results["best_comments"])
def test_highest_paid_actors1(self):
result = lib.procesar_queries([
"highest_paid_actors",
[
"filter_by_date",
["popular_movies", ["load_database"], 50, 100, "RT"], 2000
]
])
self.assertIsInstance(result, list)
self.assertEqual(result, self.results["highest_paid_actors1"])
def test_successfulll_actors2(self):
result = lib.procesar_queries([
"successful_actors",
[
"filter_by_date",
["popular_movies", ["load_database"], 60, 100, "RT"], 2001
]
])
self.assertIsInstance(result, list)
self.assertEqual(result, self.results["successful_actors2"])
def test_popular_genre(self):
result = lib.procesar_queries([
"popular_genre",
[
"filter_by_date",
["popular_movies", ["load_database"], 50, 100, "RT"], 2000,
False
], "RT"
])
self.assertIsInstance(result, list)
self.assertEqual(result, self.results["popular_genre"])
def obtener_resultado(query):
try:
result = lib.procesar_queries(query)
except (BadQuery, WrongInput, MovieError) as err:
result = err
return result
def test_popular_actors(self):
result = lib.procesar_queries(
["popular_actors", ["load_database"], 1, 10, "RT"])
self.assertIsInstance(result, list)
self.assertEqual(result, self.results["popular_actors"])
def test_take_movie_while(self):
result = lib.procesar_queries(
["take_movie_while", ["load_database"], "date", "<", 2005])
movies = {int(movie.id) for movie in result}
self.assertEqual(movies, self.results["take_movie_while"])
def test_load_database(self):
result = lib.procesar_queries(["load_database"])
movies = {int(movie.id) for movie in result}
self.assertEqual(movies, self.results["load_database"])
