def test_repository_can_add_movie(in_memory_repo): movie = Movie('Finding Potato', 2020) movie.rank = 6 in_memory_repo.add_movie(movie) assert in_memory_repo.get_movie(6) is movie assert in_memory_repo.get_movie(6).rank == 6
def read_csv_file(self): with open(self.__file_name, mode='r', encoding='utf-8-sig') as csvfile: movie_file_reader = csv.DictReader(csvfile) for row in movie_file_reader: movie = Movie(row["Title"], int(row["Year"])) movie.rank = int(row["Rank"]) movie.description = str(row["Description"]) director = Director(row["Director"]) actors = row["Actors"].split(",") genres = row["Genre"].split(",") if movie not in self.__dataset_of_movies: self.__dataset_of_movies.append(movie) if director not in self.__dataset_of_directors: self.__dataset_of_directors.append(director) for i in actors: actor = Actor(i) if actor not in self.__dataset_of_actors: self.__dataset_of_actors.append(actor) for i in genres: genre = Genre(i) if genre not in self.__dataset_of_genres: self.__dataset_of_genres.append(genre)