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)
