def test_manchester_united():
fixture = read_fixture('manchester_united.json')
source = FootballWikiSource()
source.set_content(fixture)
models = list(source.get_models())
assert len(models) == 1
team = models[0]
print(team)
assert team.get_node_name() == 'Manchester_United_F_C:SportsTeam'
assert team.get_type() == 'SportsTeam'
assert team.get_name(
) == 'Manchester United F.C.', 'name should be taken from page title to allow proper linking'
assert team.get_property('sport') == "Football"
assert team.get_property('foundingDate') == 1878
assert team.get_property('ground') == 'Old Trafford'
assert team.get_property('memberOf') == 'Premier League'
assert team.get_property('url') == 'http://www.manutd.com/'
assert team.get_relation_targets('coach') == ['Ole_Gunnar_Solskjr:Person']
assert len(team.get_relation_targets('athlete')) == 29
assert team.get_relation_targets('athlete')[0] == ('David_de_Gea:Person', {
'number': 1,
'position': 'GK'
}), 'This relation should have a property'
assert team.get_relation_targets('athlete')[2] == ('Eric_Bailly:Person', {
'number': 3,
'position': 'DF'
}), 'This relation should have a property'
def test_zlatan():
fixture = read_fixture('zlatan.json')
source = FootballWikiSource()
source.set_content(fixture)
models = list(source.get_models())
assert len(models) == 1
zlatan = models[0]
print(zlatan)
assert zlatan.get_node_name() == 'Zlatan_Ibrahimovi:Person'
assert zlatan.get_type() == 'Person'
assert zlatan.get_name() == 'Zlatan Ibrahimović'
assert zlatan.get_property('birthDate') == 1981
assert zlatan.get_property('birthPlace') == 'Malmö'
assert zlatan.get_property('nationality') == 'Sweden'
assert zlatan.get_property('height') == 1.95
assert zlatan.get_relation_targets('athlete')[0] == ('Malm_FF:SportsTeam',
{
'since': 1999,
'until': 2001
})
assert zlatan.get_relation_targets('athlete')[-1] == (
'LA_Galaxy:SportsTeam', {
'since': 2018
})
assert zlatan.get_relation_targets('coach') is None
def test_allegri():
fixture = read_fixture('massimiliano_allegri.json')
source = FootballWikiSource()
source.set_content(fixture)
models = list(source.get_models())
assert len(models) == 1
coach = models[0]
print(coach)
assert coach.get_node_name() == 'Massimiliano_Allegri:Person'
assert coach.get_type() == 'Person'
assert coach.get_name() == 'Massimiliano Allegri'
assert coach.get_property('birthDate') == 1967
assert coach.get_property('birthPlace') == 'Livorno'
assert coach.get_property('nationality') == 'Italy'
assert coach.get_property('height') == 1.83
assert coach.get_relation_targets('athlete')[-1] == (
'Aglianese_Calcio_1923:SportsTeam', {
'since': 2001,
'until': 2003
})
assert coach.get_relation_targets('coach')[-1] == (
'Juventus_F_C:SportsTeam', {
'since': 2014
})
def test_burnley():
fixture = read_fixture('burnley.json')
source = FootballWikiSource()
source.set_content(fixture)
models = list(source.get_models())
assert len(models) == 1
team = models[0]
print(team)
assert team.get_node_name() == 'Burnley_F_C:SportsTeam'
assert team.get_type() == 'SportsTeam'
assert team.get_name(
) == 'Burnley F.C.', 'name should be taken from page title to allow proper linking'
assert team.get_relation_targets('coach') == ['Sean_Dyche:Person']
assert len(team.get_relation_targets('athlete')) == 37
def test_extract_clubs_and_years():
template = Template(
page_title='Foo',
name='Bar',
parameters={
'years':
" 1990–1996<br>1996<br>1996–2007<br>'''Total''' ",
'clubs':
" FlagiconNOR [[Clausenengen FK|Clausenengen]]<br>FlagiconNOR [[Molde FK|Molde]]<br>FlagiconENG [[Manchester United F.C.|Manchester United]] ",
})
relations = list(
FootballWikiSource.extract_clubs_and_years(template, 'clubs', 'years'))
print(relations)
assert len(relations) == 3
assert relations[0] == ('Clausenengen FK', (1990, 1996))
assert relations[1] == ('Molde FK', (1996, 1996)), 'A single year contract'
assert relations[2] == ('Manchester United F.C.', (1996, 2007))
def test_extract_clubs_and_years_open_period():
template = Template(
page_title='Foo',
name='Bar',
parameters={
'years':
" 2011-2012<br>2012-2016<br>2016–2018<br>2017–2018<br>2018- ",
'clubs':
" flagiconITA [[A.C. Milan|Milan]]<br>flagiconFRA [[Paris Saint-Germain F.C.|Paris Saint-Germain]]<br>flagiconENG [[Manchester United]]<br>flagiconENG [[Manchester United]]<br>flagiconUSA [[LA Galaxy] ",
})
relations = list(
FootballWikiSource.extract_clubs_and_years(template, 'clubs', 'years'))
print(relations)
assert len(relations) == 5
assert relations[0] == ('A.C. Milan', (2011, 2012))
assert relations[1] == ('Paris Saint-Germain F.C.', (2012, 2016))
assert relations[2] == ('Manchester United', (2016, 2018))
assert relations[3] == ('Manchester United', (2017, 2018))
assert relations[4] == ('LA Galaxy',
(2018, None)), 'Properly parses and open period'
def test_ole_gunnar():
fixture = read_fixture('ole_gunnar.json')
source = FootballWikiSource()
source.set_content(fixture)
assert len(
source.get_templates_of_type(template_type='Country flagicon2')) == 17
assert len(
source.get_templates_of_type(template_type='Birth date and age')) == 1
models = list(source.get_models())
assert len(models) == 1
ole = models[0]
print(ole)
assert ole.get_node_name() == 'Ole_Gunnar_Solskjr:Person'
assert ole.get_type() == 'Person'
assert ole.get_name() == 'Ole Gunnar Solskjær'
assert ole.get_property('birthDate') == 1973
assert ole.get_property('birthPlace') == 'Kristiansund'
assert ole.get_property('nationality') == 'Norway'
assert ole.get_property('height') == 1.78
assert ole.get_property('foo') is None
assert len(ole.get_relation_targets('athlete')) == 3
assert ole.get_relation_targets('athlete')[0] == (
'Clausenengen_FK:SportsTeam', {
'since': 1990,
'until': 1994
})
assert ole.get_relation_targets('athlete')[1] == ('Molde_FK:SportsTeam', {
'since': 1994,
'until': 1996
})
assert ole.get_relation_targets('athlete')[2] == (
'Manchester_United_F_C:SportsTeam', {
'since': 1996,
'until': 2007
})
assert len(ole.get_relation_targets('coach')) == 6
assert ole.get_relation_targets('coach')[0] == (
'Manchester_United_F_C_Reserves_and_Academy:SportsTeam', {
'since': 2008,
'until': 2011
})
assert ole.get_relation_targets('coach')[1] == ('Molde_FK:SportsTeam', {
'since': 2011,
'until': 2014
})
assert ole.get_relation_targets('coach')[2] == (
'Cardiff_City_F_C:SportsTeam', {
'since': 2014,
'until': 2014
}), 'A single year contract'
assert ole.get_relation_targets('coach')[3] == (
'Clausenengen_FK:SportsTeam', {
'since': 2014,
'until': 2016
})
assert ole.get_relation_targets('coach')[4] == ('Molde_FK:SportsTeam', {
'since': 2015,
'until': 2018
})
assert ole.get_relation_targets('coach')[5] == (
'Manchester_United_F_C:SportsTeam', {
'since': 2018,
'until': 2019
})
assert ole.get_relation_targets('foo') is None