def test_runtime(): movie1 = Movie("Moana", 2000) assert movie1.runtime_minutes == 0 movie1.runtime_minutes = "3" assert movie1.runtime_minutes == 0 movie1.runtime_minutes = None assert movie1.runtime_minutes == 0 try: movie1.runtime_minutes = 0 except ValueError: pass else: raise AssertionError assert movie1.runtime_minutes == 0 try: movie1.runtime_minutes = -4 except ValueError: pass else: raise AssertionError assert movie1.runtime_minutes == 0 movie1.runtime_minutes = 12 assert movie1.runtime_minutes == 12
def load_movies(data_path, repo): filename = os.path.join(data_path, "movies.csv") with open(filename, mode='r', encoding='utf-8-sig') as csvfile: movie_file_reader = csv.DictReader(csvfile) for row in movie_file_reader: media_id = row["Id"] title = row['Title'] release_year = int(row['Year']) description = row["Description"] runtime_minutes = row["Runtime (Minutes)"] director = Director(row["Director"]) genres = set([Genre(x) for x in row["Genre"].split(',')]) actors = set([Actor(x) for x in row["Actors"].split(',')]) movie = Movie(title, release_year) movie.description = description movie.runtime_minutes = runtime_minutes movie.director = director movie.media_id = media_id for genre in genres: movie.add_genre(genre) for actor in actors: movie.add_actor(actor) repo.add_media(movie)
def test_init(self): movie = Movie("Moana", 2016, 7) print(movie) director = Director("Ron Clements") movie.director = director print(movie.director) actors = [Actor("Auli'i Cravalho"), Actor("Dwayne Johnson"), Actor("Rachel House"), Actor("Temuera Morrison")] for actor in actors: movie.add_actor(actor) assert movie.actors == [Actor("Auli'i Cravalho"), Actor("Dwayne Johnson"), Actor("Rachel House"), Actor("Temuera Morrison")] assert len(movie.actors) == 4 movie.runtime_minutes = 107 print("Movie runtime: {} minutes".format(movie.runtime_minutes))
def movie(): movie = Movie("Moana", 2000) movie.runtime_minutes = 12 return movie