def test_extract_credits(): credits_data = load_test_data('title_maindetails.json')['credits'] data = {'credits': credits_data[0:3]} title = Title(data=data) people = title._extract_credits(data) assert 6 == len(people) assert people[0].roles == [] assert people[0].label == 'Directed by' assert people[0].token == 'directors' assert people[0].name == 'Aleksey Popogrebskiy' assert people[0].imdb_id == 'nm1403225' assert people[5].roles == ['Golos po ratsii - Volodya'] assert people[5].label == 'Cast' assert people[5].token == 'cast' assert people[5].name == 'Ilya Sobolev' assert people[5].imdb_id == 'nm3777825' assert people[2].roles == ['Sergey'] assert people[2].label == 'Cast' assert people[2].token == 'cast' assert people[2].name == 'Sergey Puskepalis' assert people[2].imdb_id == 'nm1655234'
def set_up(): data = load_test_data('title_maindetails.json') title = Title(data=data) return {'title_data': data, 'title': title}
def test_extract_year_no_year(): data = {'year': '????'} title = Title(data=data) assert title._extract_year(data) is None