def test_completed_attributes():
filename = 'datafiles/Data1000Movies.csv'
movie_file_reader = MovieFileCSVReader(filename)
movie_file_reader.read_csv_file()
movie_name = "Guardians of the Galaxy"
movie = movie_file_reader.movie_object(movie_name)
# normal attributes
assert movie.title == "Guardians of the Galaxy"
assert movie.release_year == 2014
assert movie.description == "A group of intergalactic criminals are forced to work together " \
"to stop a fanatical warrior from taking control of the universe."
assert movie.director == Director("James Gunn")
assert movie.actors == [
Actor("Chris Pratt"),
Actor("Vin Diesel"),
Actor("Bradley Cooper"),
Actor("Zoe Saldana")
]
assert movie.genres == [
Genre("Action"), Genre("Adventure"),
Genre("Sci-Fi")
]
assert movie.runtime_minutes == 121
# extra attributes
assert movie.rank == 1
assert movie.rating == 8.1
assert movie.revenue == 333.13
assert movie.votes == 757074
assert movie.metascore == 76
def test_incomplete_attributes():
filename = 'datafiles/Data1000Movies.csv'
movie_file_reader = MovieFileCSVReader(filename)
movie_file_reader.read_csv_file()
movie_name = "Hounds of Love"
movie = movie_file_reader.movie_object(movie_name)
# normal attributes
assert movie.title == "Hounds of Love"
assert movie.release_year == 2016
assert movie.description == "A cold-blooded predatory couple while cruising the streets in search of their " \
"next victim, will stumble upon a 17-year-old high school girl, who will be sedated, " \
"abducted and chained in the strangers' guest room."
assert movie.director == Director("Ben Young")
assert movie.actors == [
Actor("Emma Booth"),
Actor("Ashleigh Cummings"),
Actor("Stephen Curry"),
Actor("Susie Porter")
]
assert movie.genres == [Genre("Crime"), Genre("Drama"), Genre("Horror")]
assert movie.runtime_minutes == 108
# extra attributes
assert movie.rank == 23
assert movie.rating == 6.7
assert movie.revenue is None
assert movie.votes == 1115
assert movie.metascore == 72
