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