def test_director(): movie1 = Movie("Moana", 2000) assert movie1.director == Director(None) movie1.director = Director("Ron Clements") assert movie1.director == Director("Ron Clements") movie1.director = Director("Rob Clements") assert movie1.director == Director("Rob Clements")
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_repository_does_not_retrieve_a_non_existent_director(): mem_repo = MemoryRepository() up_movie = Movie("Up", 2009, 1) inside_out_movie = Movie("Inside Out", 2015, 2) dolittle_movie = Movie("Dolittle", 2019, 3) up_movie.director = Director("Pete Docter") inside_out_movie.director = Director("Pete Docter") dolittle_movie.director = Director("Stephen Gaghan") mem_repo.add_movie(up_movie) mem_repo.add_movie(inside_out_movie) mem_repo.add_movie(dolittle_movie) movies_by_director = mem_repo.get_movies_by_director( Director("Sergio Pablos")) assert len(movies_by_director) == 0 assert movies_by_director == []
def test_repository_can_get_movies_by_director(): mem_repo = MemoryRepository() up_movie = Movie("Up", 2009, 1) inside_out_movie = Movie("Inside Out", 2015, 2) dolittle_movie = Movie("Dolittle", 2019, 3) up_movie.director = Director("Pete Docter") inside_out_movie.director = Director("Pete Docter") dolittle_movie.director = Director("Stephen Gaghan") mem_repo.add_movie(up_movie) mem_repo.add_movie(inside_out_movie) mem_repo.add_movie(dolittle_movie) movies_by_director = mem_repo.get_movies_by_director( Director("Pete Docter")) assert len(movies_by_director) == 2 assert movies_by_director == [inside_out_movie, up_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