def test_movie_less_than_operator(): movie_1 = Movie("Project X", 2012) movie_2 = Movie("ABC", 2018) movie_3 = Movie("Project X", 2019) assert movie_2 < movie_1 < movie_3
def test_can_get_movie(in_memory_repo): movie = Movie("Guardians of the Galaxy", 2014) result_movie = services.get_movie(in_memory_repo, movie.title, movie.year) assert result_movie.id == 1 assert result_movie.title == "Guardians of the Galaxy" assert result_movie.year == 2014 assert result_movie.director == Director("James Gunn") assert result_movie.runtime_minutes == 121
def read_csv_file(self): with open(self._filename, mode='r', encoding='utf-8-sig') as csvfile: reader = csv.DictReader(csvfile) index = 2 for row in reader: title = row["Title"] release_year = int(row["Year"]) movie1 = Movie(title, release_year) self._dataset_of_movies.append(movie1) movie1.description = row["Description"] movie1.runtime_minutes = int(row["Runtime (Minutes)"]) movie1.id = int(row["Rank"]) director_name = row["Director"] director = Director(director_name) if director not in self._dataset_of_directors: self._dataset_of_directors.append(director) movie1.director = director else: n = self._dataset_of_directors.index(director) movie1.director = self._dataset_of_directors[n] director.movie_list.append(movie1) actor_name_list = row["Actors"].split(",") index1 = 1 for name in actor_name_list: actor = Actor(name.strip()) if actor not in self._dataset_of_actors: self._dataset_of_actors.append(actor) movie1.actors.append(actor) actor.act_movie.append(movie1) else: n = self._dataset_of_actors.index(actor) movie1.actors.append(self._dataset_of_actors[n]) self._dataset_of_actors[n].act_movie.append(movie1) index1 += 1 for actor in movie1.actors: for colleague in movie1.actors: if actor != colleague: actor.add_actor_colleague(colleague) genre_list = row["Genre"].split(",") for type in genre_list: genre = Genre(type.strip()) movie1.genres.append(genre) if genre not in self._dataset_of_genres: self._dataset_of_genres.append(genre) movie1.rating = row["Votes"] movie1.metascore = row["Metascore"] index += 1
def movie(): movie = Movie("Project X", 2012) movie.id = 1 movie.director = Director("Mike John") movie.runtime_minutes = 120 movie.description = "abcdefg" return movie
def review(): return Review(Movie("Project X", 2012), "A nice Movie", 7 ,User('dbowie', '1234567890'))
def test_repository_can_add_movie(in_memory_repo): movie = Movie("ABC", 2020) movie.id = 1001 in_memory_repo.add_movie(movie) assert in_memory_repo.get_movie_by_id(1001) is movie
def get_movie(repo: AbstractRepository, title: str, year: int) -> Movie: return repo.get_movie(Movie(title, year))