def test_init(): movie = Movie("Moana", 2016) assert repr(movie) == "<Movie Moana, 2016>" assert movie.title == "Moana" movie1 = Movie("", "1") assert repr(movie1) == "<Movie None, None>" director = Director("Ron Clements") movie.director = director assert repr(movie.director) == "<Director Ron Clements>" movie.description = "Moana, daughter of chief Tui, embarks on a journey to return the heart of goddess Te Fitti from Maui, a demigod, after the plants and the fish on her island start dying due to a blight." assert movie.description == "Moana, daughter of chief Tui, embarks on a journey to return the heart of goddess Te Fitti from Maui, a demigod, after the plants and the fish on her island start dying due to a blight." actors = [ Actor("Auli'i Cravalho"), Actor("Dwayne Johnson"), Actor("Rachel House"), Actor("Temuera Morrison") ] for actor in actors: movie.add_actor(actor) assert repr( movie.actors ) == "[<Actor Auli'i Cravalho>, <Actor Dwayne Johnson>, <Actor Rachel House>, <Actor Temuera Morrison>]" movie.runtime_minutes = 107 assert repr(movie.runtime_minutes) == "107"
def test_equal(): actor1 = Actor("Angelina Jolie") actor2 = Actor("Brad Pitt") actor3 = Actor("Angelina Jolie") assert (actor1 == actor2) == False assert (actor1 == actor3) == True
def read_csv_file(self, file_name): with open(file_name, mode='r', encoding='utf-8-sig') as csvfile: movie_file_reader = csv.DictReader(csvfile) for row in movie_file_reader: id = int(row['Rank']) title = row['Title'] release_year = int(row['Year']) movie = Movie(title, release_year, id) self.add_movie(movie) actors = row["Actors"] actors = actors.split(",") for actor in actors: actor = Actor(actor) self.add_actor(actor) actor = self.get_actor(actor.actor_full_name) movie.add_actor(actor) actor.add_movie(movie) director = Director(row["Director"]) self.add_director(director) director = self.get_director(director.director_full_name) movie.director = director director.add_movie(movie) genres = row["Genre"] genres = genres.split(",") for genre in genres: genre = Genre(genre) movie.add_genre(genre) self.add_genre(genre) movie.description = row["Description"] movie.runtime_minutes = int(row["Runtime (Minutes)"])
def test_init(): actor1 = Actor("Angelina Jolie") assert repr(actor1) == "<Actor Angelina Jolie>" actor2 = Actor("") assert repr(actor2) == "<Actor None>" actor3 = Actor(42) assert repr(actor3) == "<Actor None>"
def test_movie_actors(): # correct input movie1 = Movie("Moana", 2016) movie1.actors = [Actor("Dwayne Johnson"), Actor("Rachel House")] assert movie1.actors == [Actor("Dwayne Johnson"), Actor("Rachel House")] # incorrect input movie2 = Movie("Frozen", 2014)
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"]) movie = Movie(title, release_year) movie.description = row["Description"] director = row["Director"] movie.add_director(Director(director)) if Director(director) not in self.__dataset_of_directors: self.__dataset_of_directors.append(Director(director)) actors = row["Actors"].split(",") for actor in actors: movie.add_actor(Actor(actor)) if Actor(actor) not in self.__dataset_of_actors: self.__dataset_of_actors.append(Actor(actor)) genres = row["Genre"].split(",") for genre in genres: movie.add_genre(Genre(genre)) if Genre(genre) not in self.__dataset_of_genres: self.__dataset_of_genres.append(Genre(genre)) movie.runtime_minutes = int(row["Runtime (Minutes)"]) try: movie.external_rating = float(row["Rating"]) except ValueError: pass try: movie.rating_votes = int(row["Votes"]) except ValueError: pass try: movie.revenue_in_millions = float( row["Revenue (Millions)"]) except ValueError: pass try: movie.metascore = int(row["Metascore"]) except ValueError: pass self.__dataset_of_movies.append(movie) # title = row['Title'] # release_year = int(row['Year']) # print(f"Movie {index} with title: {title}, release year {release_year}") index += 1
def test_add_colleague(): actor1 = Actor("Angelina Jolie") actor2 = Actor("Brad Pitt") assert actor1.check_if_this_actor_worked_with(actor2) == False actor1.add_actor_colleague(actor2) assert actor1.check_if_this_actor_worked_with(actor2) == True assert actor2.check_if_this_actor_worked_with(actor1) == True
def browse_by_actor(): actorsMovies = movieByActor() random_movies = repo.repo_instance.get_random_movies(4) random_actors = repo.repo_instance.get_random_actors(9) if actorsMovies.validate_on_submit(): post_actor = Actor(actorsMovies.actor_name.data) movies = repo.repo_instance.get_movies_acted_by(post_actor) if movies != []: return render_template( 'movies/actors_movies.html', movies = movies, actor = post_actor, movieSearch = movieByTitle(), ) else: return render_template( 'movies/actors_movies.html', movies = movies, actor = None, movieSearch = movieByTitle(), ) return render_template( 'movies/get_actor.html', random_movies = random_movies, random_actors = random_actors, movieSearch = movieByTitle(), actorSearch = actorsMovies, )
def test_repository_can_add_and_get_actors(in_memory_repo): actor = Actor("Matt Damon") in_memory_repo.add_actor(actor) assert (actor in in_memory_repo.get_actors()) is True assert len(in_memory_repo.get_actors()) == 321
def test_actor_check_if_this_actor_worked_with(): actor_1 = Actor("Angelina Jolie") actor_2 = Actor("Tom Cruise") actor_3 = Actor("Chris Pratt") actor_1.add_actor_colleague(actor_2) assert actor_1.check_if_this_actor_worked_with(actor_2) is True assert actor_1.check_if_this_actor_worked_with(actor_3) is False
def test_actor(): movie = Movie("Moana", 2016) actors = [ Actor("Auli'i Cravalho"), Actor("Dwayne Johnson"), Actor("Rachel House"), Actor("Temuera Morrison") ] for actor in actors: movie.add_actor(actor) movie.add_actor("fake actor") assert repr( movie.actors ) == "[<Actor Auli'i Cravalho>, <Actor Dwayne Johnson>, <Actor Rachel House>, <Actor Temuera Morrison>]" movie.actors = [] assert repr(movie.actors) == "[]" movie.actors = [1, 2, 3] assert repr(movie.actors) == "[]"
def test_movie_add_actor(): # correct input movie1 = Movie("Moana", 2016) movie1.add_actor(Actor("Dwayne Johnson")) assert movie1.actors == [Actor("Dwayne Johnson")] # incorrect input movie1.add_actor(Director("Ron Clements")) assert movie1.actors == [Actor("Dwayne Johnson")] # adding an actor twice movie1.add_actor(Actor("Dwayne Johnson")) assert movie1.actors == [Actor("Dwayne Johnson")] # adding a second actor movie1.add_actor(Actor("Rachel House")) assert movie1.actors == [Actor("Dwayne Johnson"), Actor("Rachel House")]
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']) movie = Movie(title, release_year) movie.description = row['Description'] movie.director = Director(row['Director']) if Director( row['Director']) not in self.__dataset_of_directors: self.__dataset_of_directors.append( Director(row['Director'])) for actor in row['Actors'].split(','): movie.add_actor(Actor(actor)) if Actor(actor) not in self.dataset_of_actors: self.dataset_of_actors.append(Actor(actor)) for genre in row['Genre'].split(','): movie.add_genre(Genre(genre)) if Genre(genre) not in self.dataset_of_genres: self.dataset_of_genres.append(Genre(genre)) movie.runtime_minutes = int(row['Runtime (Minutes)']) try: movie.revenue = float(row['Revenue (Millions)']) except: pass self.__dataset_of_movies.append(movie)
def test_movie_remove_actor(): movie1 = Movie("Moana", 2016) movie1.add_actor(Actor("Dwayne Johnson")) # correct input movie1.remove_actor(Actor("Dwayne Johnson")) assert movie1.actors == [] # incorrect input movie1.add_actor(Actor("Dwayne Johnson")) movie1.remove_actor(Director("Dwayne Johnson")) assert movie1.actors == [Actor("Dwayne Johnson")] # removing an actor not in list movie1.remove_actor(Actor("Rachel House")) assert movie1.actors == [Actor("Dwayne Johnson")]
def test_hash(): actor1 = Actor("Angelina Jolie") assert hash(actor1) == hash("Angelina Jolie") actor2 = Actor("Brad Pitt") assert hash(actor2) == hash("Brad Pitt")
def test_repository_can_add_a_actor(in_memory_repo): actor = Actor("Dave Brown") in_memory_repo.add_actor(actor) assert actor in in_memory_repo.get_actors()
def test_repository_can_retrieve_actors(in_memory_repo): actors = in_memory_repo.get_actors() assert len(actors) == 20 assert Actor("Chris Pratt") in actors
def test_repository_can_add_a_actor(session_factory): repo = SqlAlchemyRepository(session_factory) actor = Actor("Dave Brown") repo.add_actor(actor) assert actor in repo.get_actors()
def test_repository_can_retrieve_actors(session_factory): repo = SqlAlchemyRepository(session_factory) actors = repo.get_actors() assert len(actors) == 2394 assert Actor("Chris Pratt") in actors
def load_movies_and_genres(data_path: str, repo: MemoryRepository): genres = dict() directors = dict() actors = dict() for data_row in read_csv_file(os.path.join(data_path, 'Data1000Movies.csv')): movie_key = int(data_row[0]) movie_genres = data_row[2].lower().split(",") movie_directors = data_row[4].lower().split(",") movie_actors = data_row[5].lower().split(",") # Add any new genres; associate the current movie with genres. for genre in movie_genres: if genre not in genres.keys(): genres[genre] = list() genres[genre].append(movie_key) for director in movie_directors: if director not in directors.keys(): directors[director] = list() directors[director].append(movie_key) for actor in movie_actors: if actor not in actors.keys(): actors[actor] = list() actors[actor].append(movie_key) # Create Movie object. movie = Movie(title=data_row[1].lower(), year=int(data_row[6]), movie_id=int(data_row[0])) movie.description = data_row[3] movie.runtime_minutes = int(data_row[7]) movie.rating = float(data_row[8]) movie.votes = float(data_row[9]) if data_row[10] != 'N/A': movie.revenue = f"{data_row[10]}Millions" if data_row[11] != 'N/A': movie.metascore = data_row[11] # Add the Movie to the repository. repo.add_movie(movie) # Create Genre objects, associate them with Movies and add them to the repository. for genre_name in genres.keys(): genre = Genre(genre_name) for movie_id in genres[genre_name]: movie = repo.get_movie(movie_id) make_genre_association(movie, genre) repo.add_genre(genre) for director_name in directors.keys(): director = Director(director_name) for movie_id in directors[director_name]: movie = repo.get_movie(movie_id) make_director_association(movie, director) repo.add_director(director) for actor_name in actors.keys(): actor = Actor(actor_name) for movie_id in actors[actor_name]: movie = repo.get_movie(movie_id) make_actor_association(movie, actor) repo.add_actor(actor)
def test_repository_add_actor(in_memory_repo): actor = Actor("Shah Rukh Khan") in_memory_repo.add_actor(actor) assert actor in in_memory_repo.get_actors()
def test_actor_add_actor_colleague(): actor_1 = Actor("Angelina Jolie") actor_2 = Actor("Tom Cruise") # add new object not of type Actor actor_1.add_actor_colleague(50) assert len(actor_1.colleagues) == 0 # add new colleague actor_1.add_actor_colleague(actor_2) assert actor_1.colleagues[0] == actor_2 # add existing colleague actor_1.add_actor_colleague(actor_2) assert len(actor_1.colleagues) == 1 # add a second colleague actor_3 = Actor("Chris Pratt") actor_1.add_actor_colleague(actor_3) assert len(actor_1.colleagues) == 2 assert actor_1.colleagues[1] == actor_3
def actor(): return Actor("Chris Pratt")
def actor2(): return Actor("Matt Damon")
def test_less_than(): actor1 = Actor("Angelina Jolie") actor2 = Actor("Brad Pitt") assert (actor1 < actor2) == True
def actor(): return Actor("Jack Black")
def test_actor_equals_operator(): actor_1 = Actor("Angelina Jolie") actor_2 = Actor("Jack Black") actor_3 = Actor("Angelina Jolie") assert (actor_1 == actor_2) is False assert (actor_1 == actor_3) is True
def test_add_and_get_actor(in_memory_repo): in_memory_repo.add_actor(Actor("new actor")) assert Actor("new actor") in in_memory_repo.get_actors() assert Actor("fake actor") not in in_memory_repo.get_actors()
def test_actor_less_than_operator(): actor_1 = Actor("Angelina Jolie") actor_2 = Actor("Jack Black") assert (actor_1 < actor_2) is True assert (actor_2 < actor_1) is False
def test_actor_hash(): actor_1 = Actor("Angelina Jolie") actor_2 = Actor("Jack Black") actor_3 = Actor("Angelina Jolie") assert actor_1.__hash__() != actor_2.__hash__() assert actor_1.__hash__() == actor_3.__hash__()