Ejemplo n.º 1
0
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
Ejemplo n.º 2
0
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)
Ejemplo n.º 3
0
    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))
Ejemplo n.º 4
0
def movie():
    movie = Movie("Moana", 2000)
    movie.runtime_minutes = 12
    return movie