def load_movies(data_path, repo): filename = os.path.join(data_path, "movies.csv") with open(filename, mode='r', encoding='utf-8-sig') as csvfile: movie_file_reader = csv.DictReader(csvfile) for row in movie_file_reader: media_id = row["Id"] title = row['Title'] release_year = int(row['Year']) description = row["Description"] runtime_minutes = row["Runtime (Minutes)"] director = Director(row["Director"]) genres = set([Genre(x) for x in row["Genre"].split(',')]) actors = set([Actor(x) for x in row["Actors"].split(',')]) movie = Movie(title, release_year) movie.description = description movie.runtime_minutes = runtime_minutes movie.director = director movie.media_id = media_id for genre in genres: movie.add_genre(genre) for actor in actors: movie.add_actor(actor) repo.add_media(movie)
def test_init(self): movie = Movie("Moana", 2016, 7) print(movie) director = Director("Ron Clements") movie.director = director print(movie.director) actors = [Actor("Auli'i Cravalho"), Actor("Dwayne Johnson"), Actor("Rachel House"), Actor("Temuera Morrison")] for actor in actors: movie.add_actor(actor) assert movie.actors == [Actor("Auli'i Cravalho"), Actor("Dwayne Johnson"), Actor("Rachel House"), Actor("Temuera Morrison")] assert len(movie.actors) == 4 movie.runtime_minutes = 107 print("Movie runtime: {} minutes".format(movie.runtime_minutes))
def read_csv_file(self): with open(self.__file_name, mode='r', encoding='utf-8-sig') as csvfile: movie_file_reader = csv.DictReader(csvfile) index = 0 for row in movie_file_reader: title = row['Title'] release_year = int(row['Year']) actors = row['Actors'].split(",") genres = row['Genre'].split(",") director = Director(row['Director']) rating = row['Rating'] votes = row['Votes'] revenue = row['Revenue (Millions)'] metascore = row['Metascore'] movie = Movie(title, release_year, index + 1) movie.director = director # extension attribute application movie.external_rating = rating movie.rating_votes = votes movie.revenue = revenue movie.metascores = metascore # adding the actors, genres, director and the movie to the corresponding datasets for i in range(len(actors)): actor = Actor(actors[i]) movie.add_actor(actor) if actor not in self.__dataset_of_actors: # adding only unique actors to the dataset self.__dataset_of_actors.add(actor) for i in range(len(genres)): genre = Genre(genres[i]) movie.add_genre(genre) if genre not in self.__dataset_of_genres: # adding only unique genres to the dataset self.__dataset_of_genres.add(genre) if movie.director not in self.__dataset_of_directors: # adding only unique directors to the dataset self.__dataset_of_directors.add(movie.director) self.__dataset_of_movies.append(movie) index += 1
def test_repository_does_not_retrieve_a_non_existent_actor(): mem_repo = MemoryRepository() up_movie = Movie("Up", 2009, 1) the_avengers_movie = Movie("The Avengers", 2012, 2) dolittle_movie = Movie("Dolittle", 2019, 3) up_movie.add_actor(Actor("Ed Asner")) the_avengers_movie.add_actor(Actor("Robert Downey, Jr.")) dolittle_movie.add_actor(Actor("Robert Downey, Jr.")) mem_repo.add_movie(up_movie) mem_repo.add_movie(the_avengers_movie) mem_repo.add_movie(dolittle_movie) movies_by_actor = mem_repo.get_movies_by_actor(Actor("Jim Carrey")) assert len(movies_by_actor) == 0 assert movies_by_actor == []
def test_repository_can_get_movies_by_actor(): mem_repo = MemoryRepository() up_movie = Movie("Up", 2009, 1) the_avengers_movie = Movie("The Avengers", 2012, 2) dolittle_movie = Movie("Dolittle", 2019, 3) up_movie.add_actor(Actor("Ed Asner")) the_avengers_movie.add_actor(Actor("Robert Downey, Jr.")) dolittle_movie.add_actor(Actor("Robert Downey, Jr.")) mem_repo.add_movie(up_movie) mem_repo.add_movie(the_avengers_movie) mem_repo.add_movie(dolittle_movie) movies_by_actor = mem_repo.get_movies_by_actor(Actor("Robert Downey, Jr.")) assert len(movies_by_actor) == 2 assert movies_by_actor == [dolittle_movie, the_avengers_movie]
def test_actors(): movie1 = Movie("Moana", 2000) actors = [Actor("Auli'i Cravalho"), Actor("Dwayne Johnson")] assert movie1.actors == set() movie1.add_actor(actors[0]) assert movie1.actors == set([Actor("Auli'i Cravalho")]) movie1.add_actor(actors[1]) assert movie1.actors == set([Actor("Auli'i Cravalho"), Actor("Dwayne Johnson")]) movie1.add_actor(actors[1]) assert movie1.actors == set([Actor("Auli'i Cravalho"), Actor("Dwayne Johnson")]) movie1.remove_actor(actors[1]) assert movie1.actors == set([Actor("Auli'i Cravalho")]) movie1.remove_actor(actors[1]) assert movie1.actors == set([Actor("Auli'i Cravalho")]) movie1.remove_actor(actors[0]) assert movie1.actors == set() movie1.remove_actor(actors[0]) assert movie1.actors == set()
def test_add_actor(self): movie = Movie("Moana", 2016, 1) movie.add_actor(Actor("Dwayne Johnson")) assert len(movie.actors) == 1 assert movie.actors == [Actor("Dwayne Johnson")]