def test_gdp_get_countries():
gdp = GdpData(DB_PATH)
countries = gdp.get_countries()
assert 247 == len(countries) # total number of countries
assert 'ABW' == countries[0].CountryCode # First item in list: Aruba
assert 'Aruba' == countries[0].CountryName
assert 'AW' == countries[0].CountryCode2
assert '533' == countries[0].UNCode
assert 'ZWE' == countries[-1].CountryCode # Last item in list: Zimbabwe
def test_gdp_filter_data():
gdp = GdpData(DB_PATH)
data = gdp.get_range(('NLD', 'ESP'), range(1990, 1992))
{'gdp': 318330511920.61, 'growth': 4.18322281189747} == data['NLD']['1990']
{'gdp': 327500328264.969, 'growth': 2.43913450352082} == data['NLD']['1991']
{'gdp': 535101248775.71, 'growth': 3.78139346477435} == data['ESP']['1990']
{'gdp': 575598537069.656, 'growth': 2.54600056787201} == data['ESP']['1991']
def test_gdp_get_all():
gdp = GdpData(DB_PATH)
data = gdp.get_all()
assert 264 == len(data.keys()) # total number of countries
total = 0
for e in data.values():
total += len(e.values())
assert 15576 == total
def test_gdp_get_countries_mapping():
gdp = GdpData(DB_PATH)
countries, countries2 = gdp.get_countries_mapping()
assert 247 == len(countries.keys()) # total number of countries
assert 'ABW' == countries["ABW"].CountryCode # First item in list: Aruba
assert 'Aruba' == countries["ABW"].CountryName
assert 'AW' == countries["ABW"].CountryCode2
assert '533' == countries["ABW"].UNCode
assert 'ZWE' == countries["ZWE"].CountryCode # Last item in list: Zimbabwe
#mapping 2 char country code -> 3 char country code
assert 'AW' == countries2["AW"].CountryCode2 # First item in list: Aruba
def test_fao_get_all():
gdpdata = GdpData(DB_PATH)
faodata = FaoData(FAO_DB_PATH, gdpdata)
data = faodata.get_all()
assert 174 == len(data) # total number of countries
#some data points
assert 31 == data["NLD"]["2805"]["1961"]
def test_fao_filter_data():
gdpdata = GdpData(DB_PATH)
faodata = FaoData(FAO_DB_PATH, gdpdata)
data = faodata.get_range(('NLD', ), ("2805", ), range(1961, 1962))
assert 1 == len(data["NLD"]) # total number of items
#some data points
assert 31 == data["NLD"]["2805"]["1961"]
def test_fao_filter_mixmaxmean():
gdpdata = GdpData(DB_PATH)
faodata = FaoData(FAO_DB_PATH, gdpdata)
data = faodata.get_range_max(('NLD', ), ("2805", ), range(1961, 2013))
assert ('NLD', '2805', '1996', 101) == data
data = faodata.get_range_min(('NLD', ), ("2805", ), range(1961, 2013))
assert ('NLD', '2805', '1982', 24) == data
data = faodata.get_range_mean(('NLD', ), ("2805", ), range(1961, 2013))
assert 42.40384615384615 == data
def test_gdp_filter_mixmaxmean():
gdp = GdpData(DB_PATH)
data = gdp.get_range_max(('NLD', 'ESP'), range(1990, 1992), "gdp")
assert ('ESP', '1991', {'gdp': 575598537069.656, 'growth': 2.54600056787201} ) == data
data = gdp.get_range_max(('NLD', 'ESP'), range(1990, 1992), "growth")
assert ('NLD', '1990', {'gdp': 318330511920.61, 'growth': 4.18322281189747} ) == data
data = gdp.get_range_min(('NLD', 'ESP'), range(1990, 1992), "gdp")
assert ('NLD', '1990', {'gdp': 318330511920.61, 'growth': 4.18322281189747}) == data
data = gdp.get_range_min(('NLD', 'ESP'), range(1990, 1992), "growth")
assert ('NLD', '1991', {'gdp': 327500328264.969, 'growth': 2.43913450352082}) == data
data = gdp.get_range_mean(('NLD', 'ESP'), range(1990, 1992), "gdp")
assert 439132656507 == data
data = gdp.get_range_mean(('NLD', 'ESP'), range(1990, 1992), "growth")
assert 2.75 == data
def test_fao_instantiation():
gdpdata = GdpData(DB_PATH)
faodata = FaoData(FAO_DB_PATH, gdpdata)
assert faodata is not None
def test_gdp_instantiation():
gdp = GdpData(DB_PATH)
assert gdp is not None