def test_actors(): movie1 = Movie("Spirited Away", 2000) movie2 = Movie("Your Name", 2000) movie3 = Movie("Batman", 2000) movie4 = Movie(None, None) movie1.actors = [ Actor("Jack Huang"), Actor("James Huang"), Actor("Ron Huang"), Actor("Jane Tang") ] assert movie1.actors == [ Actor("Jack Huang"), Actor("James Huang"), Actor("Ron Huang"), Actor("Jane Tang") ] movie2.actors = [ Actor("Jack Huang"), "James Huang", Actor("Ron Huang"), 123456 ] assert movie2.actors == [] movie3.actors = Actor("Jack Huang") assert movie3.actors == [Actor("Jack Huang")] movie4.actors = "Jack Huang" assert movie4.actors == []
def test_setters(self): # inherently tests getters too a_movie = Movie("Back To The Future", 1965) director1 = Director("Brad Bird") a_movie.title = " Mission Impossible: Ghost Protocol " assert a_movie.title == "Mission Impossible: Ghost Protocol" a_movie.title = 15 #assert a_movie.title == "Mission Impossible: Ghost Protocol" a_movie.description = " An action movie " assert a_movie.description == "An action movie" a_movie.director = director1 assert a_movie.director == director1 actor1 = Actor("Tom Cruise") actor2 = Actor("Simon Pegg") a_movie.actors = [actor1, actor2] assert a_movie.actors == [actor1, actor2] genre1 = Genre("Action") genre2 = Genre("Comedy") a_movie.genres = [genre1, genre2] assert a_movie.genres == [genre1, genre2] a_movie.runtime_minutes = 100 assert a_movie.runtime_minutes == 100 assert type(a_movie.runtime_minutes) == int with pytest.raises(ValueError): a_movie.runtime_minutes = 0 a_movie.runtime_minutes = -10
def test_add_actor(): movie1 = Movie("Spirited Away", 2000) movie1.actors = [Actor("James Huang")] movie1.add_actor(Actor("Jack Huang")) assert movie1.actors == [Actor("James Huang"), Actor("Jack Huang")] movie1.add_actor("Ron Huang") movie1.add_actor("") movie1.add_actor(None) movie1.add_actor(21345648) assert movie1.actors == [Actor("James Huang"), Actor("Jack Huang")]
def _read_row(self, row: _ROW) -> Movie: """ Helper method to construct a Movie from a row. Raises: ValueError: unable to parse row: {row} """ error = False try: title = row['Title'] genres = [ self._get_genre(Genre(name)) for name in row['Genre'].split(',') ] description = row['Description'] director = self._get_director(Director(row['Director'])) actors = [ self._get_actor(Actor(name)) for name in row['Actors'].split(',') ] release_year = int(row['Year']) runtime_minutes = int(row['Runtime (Minutes)']) rating = float(row['Rating']) votes = int(row['Votes']) revenue_millions = float( row['Revenue (Millions)'] ) if row['Revenue (Millions)'] != 'N/A' else None metascore = int( row['Metascore']) if row['Metascore'] != 'N/A' else None except KeyError: error = True except ValueError: error = True if error: raise ValueError(f'unable to parse row: {row}') movie = Movie(title, release_year) movie.genres = genres movie.description = description movie.director = director movie.actors = actors movie.runtime_minutes = runtime_minutes movie.rating = rating movie.votes = votes if revenue_millions: movie.revenue_millions = revenue_millions if metascore: movie.metascore = metascore return movie
def create_movie_instance(self, row): movie = Movie(row["Title"], int(row["Year"])) movie.director = Director(row["Director"]) movie.runtime_minutes = int(row["Runtime (Minutes)"]) movie.actors = [ Actor(actor.strip()) for actor in row['Actors'].split(",") ] movie.genres = [ Genre(genre.strip()) for genre in row['Genre'].split(",") ] movie.description = row["Description"] return movie
def test_remove_actor(): movie1 = Movie("Spirited Away", 2000) movie1.actors = [Actor("James Huang"), Actor("Ron Huang")] movie1.add_actor(Actor("Jack Huang")) movie1.remove_actor(Actor("Jack Huang")) assert movie1.actors == [Actor("James Huang"), Actor("Ron Huang")] movie1.remove_actor(Actor("James Huang")) movie1.remove_actor(Actor("Jawer")) movie1.remove_actor(Actor("Jackwerang")) movie1.remove_actor(Actor("Jweruang")) assert movie1.actors == [Actor("Ron Huang")] movie1.remove_actor(Actor("Ron Huang")) assert movie1.actors == []
def populated_movies(genres, directors, actors): movies = [] for i in range(10): movie = Movie(f'Movie{i}', 2000 + i) movie.genres = [genres[i]] movie.description = f'Description{i}' movie.director = directors[i] movie.actors = [actors[i]] movie.runtime_minutes = i + 1 movie.rating = float(i) movie.votes = i if i % 2 == 0: movie.revenue_millions = float(i + 1) if i % 4 == 0: movie.metascore = i * 10 movies.append(movie) return movies
def read_csv_file(self): csv_file_contents = csv.reader(open(self.__file_name, mode='r', encoding='utf-8-sig')) boolean = True for row in csv_file_contents: if boolean: boolean = False continue temp_movie = Movie("","") temp_movie.id = int(row[0])-1 temp_movie.title = row[1] genre_list = row[2].split(",") genre_list = [Genre(genre) for genre in genre_list] temp_movie.genres = genre_list temp_movie.description = row[3] temp_movie.director = Director(row[4]) actor_list = row[5].split(",") actor_list = [Actor(actor) for actor in actor_list] temp_movie.actors = actor_list temp_movie.releaseDate = int(row[6]) temp_movie.runtime_minutes = int(row[7]) temp_movie.rating = float(row[8]) temp_movie.metascore = row[11] self.movie_full_details.append(temp_movie)
def read_csv_file(self) -> None: with open(self.__filename, mode='r', encoding='utf-8-sig') as file_data: reader = csv.DictReader(file_data) self.__dataset_of_movies = list() for row in reader: new_movie = Movie(row['Title'], int(row['Year'])) new_movie.description = row['Description'] new_movie.genres = [ Genre(i.strip()) for i in row['Genre'].split(',') ] new_movie.actors = [ Actor(i.strip()) for i in row['Actors'].split(',') ] new_movie.director = Director(row['Director'].strip()) if row['Runtime (Minutes)'].isdigit(): new_movie.runtime_minutes = int(row['Runtime (Minutes)']) self.__dataset_of_movies.append(new_movie) for i in new_movie.genres: self.add_genre(i) self.__dataset_of_genres[i].append(new_movie) for i in new_movie.actors: self.add_actor(i) self.add_director(new_movie.director)
def test_init(self): movie1 = Movie("Moana", 2016) assert repr(movie1) == "<Movie Moana, 2016>" movie2 = Movie("") assert movie2.title is None assert movie2.year is None movie3 = Movie(3) assert movie3.title is None movie1.director = "er" assert movie1.director is None movie4 = Movie("Moana", 2016) movie5 = Movie("Mn", 2017) assert (movie1 == movie5) == False assert (movie1 == movie4) == True assert movie1.title == "Moana" assert movie1.description == None movie1.description = 2 assert movie1.description == None movie1.description = "adcsdv" assert movie1.description == "adcsdv" movie1.director = "as" assert movie1.director is None movie1.director = Director("asd") assert (movie1.director == Director("asd")) == True actors = [ Actor("Auli'i Cravalho"), Actor("Dwayne Johnson"), Actor("Rachel House"), Actor("Temuera Morrison") ] for actor in actors: movie1.add_actor(actor) movie6 = Movie("as", 190) genres = [Genre("asdc"), Genre("dfg")] for genre in genres: movie1.add_genre(genre) assert (movie6 == Movie("as", None)) == True assert (movie6.title == "as") == True assert movie6.year is None with pytest.raises(ValueError): movie1.runtime_minutes = -10 movie1.runtime_minutes = 10 assert movie1.runtime_minutes == 10 string1 = "a" assert (movie1 == 123) == False movie7 = Movie("a", 1900) movie8 = Movie("None", None) print(movie8.concate()) assert (movie7 < movie8) == False movie1.add_actor("sasa") assert (movie1.actors == [ Actor("Auli'i Cravalho"), Actor("Dwayne Johnson"), Actor("Rachel House"), Actor("Temuera Morrison") ]) == True movie1.remove_actor(Actor("Auli'i Cravalho")) assert (movie1.actors == [ Actor("Dwayne Johnson"), Actor("Rachel House"), Actor("Temuera Morrison") ]) == True movie1.remove_actor("asd") assert (movie1.actors == [ Actor("Dwayne Johnson"), Actor("Rachel House"), Actor("Temuera Morrison") ]) == True movie1.add_actor(Actor("Auli'i Cravalho")) assert (movie1.actors == [ Actor("Dwayne Johnson"), Actor("Rachel House"), Actor("Temuera Morrison"), Actor("Auli'i Cravalho") ]) == True movie1.remove_genre(Genre("asdc")) assert (movie1.genres == [Genre("dfg")]) == True movie1.add_genre(Genre("asdc")) assert (movie1.genres == [Genre("dfg")]) == False with pytest.raises(ValueError): movie1.runtime_minutes = "sd" movie1.actors = ['a', 'b']
def read_csv_file(self): with open(self.__file_name, mode='r', encoding='utf-8-sig') as csvfile: movie_file_reader = csv.DictReader(csvfile) self.__dataset_of_movies = list() self.__dataset_of_actors = list() self.__dataset_of_directors = list() self.__dataset_of_genres = list() index = 0 for row in movie_file_reader: title = row['Title'].strip() # stripping to safe release_year = int(row['Year']) actor_full_name_list = get_list_from_comma_string( row['Actors']) director_full_name = row['Director'].strip( ) # stripping to safe genre_name_list = get_list_from_comma_string(row['Genre']) #- other things rank = int(row['Rank']) description = row['Description'] runtime_minutes = int(row['Runtime (Minutes)']) external_rating = float(row['Rating']) external_rating_votes = int(row['Votes']) # These require try/except because some values are N/A's (preferably all would but it's SUPER ugly code) try: revenue = float(row['Revenue(Millions)']) except: revenue = None try: metascore = int(row['Metascore']) except: metascore = None #-------- Make Actor, Director and Genre Objects actor_object_list = [ Actor(actor_full_name) for actor_full_name in actor_full_name_list ] director_object = Director(director_full_name) genre_object_list = [ Genre(genre_name) for genre_name in genre_name_list ] #-------- Create and set up Movie Object movie_object = Movie(title, release_year) movie_object.actors = actor_object_list # thanks to my setter implementation movie_object.director = director_full_name movie_object.genres = genre_object_list # ^ ditto movie_object.release_year = release_year #- other things movie_object.description = description movie_object.runtime_minutes = runtime_minutes movie_object.external_rating = external_rating movie_object.external_rating_votes = external_rating_votes if revenue != None: movie_object.revenue = revenue if metascore != None: movie_object.metascore = metascore #-------- Take all the above information, and add to datasets self.__dataset_of_movies.append( movie_object ) # the assumption is that each row in the csv is a unique movie # I could create a function that adds unique things to lists only, nvm for actor_object in actor_object_list: if actor_object not in self.__dataset_of_actors: self.__dataset_of_actors.append(actor_object) if director_object not in self.__dataset_of_directors: self.__dataset_of_directors.append(director_object) for genre_object in genre_object_list: if genre_object not in self.__dataset_of_genres: self.__dataset_of_genres.append(genre_object) index += 1 # To test my assigmnent, run the following lines of code (I have copy and pasted these in my test_all.py file too) #movie_file_csv_reader_object = MovieFileCSVReader(r"C:\Users\Nathan Longhurst\OneDrive - The University of Auckland\b Comp235\Assignment\GitHub Clone (Current)\CS235FlixSkeleton\datafiles\Data1000Movies.csv") #unclean_dict = movie_file_csv_reader_object.check_csv_file() #movie_file_csv_reader_object.clean_csv_file(unclean_dict)