class TestingCountries(unittest.TestCase):
"""
Class for testing country module
"""
def setUp(self):
"""
Prescribe values for tests
"""
self.ukraine = Country('Ukraine')
self.ukraine.get_gdp('https://www.quandl.com/api/v3/'
'datasets/WWDI/UKR_NY_GDP_PCAP_CD.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
self.bulgaria = Country('Bulgaria')
self.bulgaria.get_investment_inflows('https://www.quandl.com'
'/api/v3/datasets/WWDI/'
'BGR_BX_KLT_DINV_WD_GD_ZS.json?'
'api_key=2jhCWecEKmuxzVY9ifwp')
self.hungary = Country('Hungary')
self.hungary.get_investment_outflows('https://www.quandl.com/'
'api/v3/datasets/WWDI/'
'HUN_BM_KLT_DINV_GD_ZS.json?'
'api_key=2jhCWecEKmuxzVY9ifwp')
self.poland = Country('Poland')
self.poland.get_manufacturing('https://www.quandl.com/api'
'/v3/datasets/WWDI/POL_NV_'
'IND_MANF_ZS.json?api_key'
'=2jhCWecEKmuxzVY9ifwp')
def test_gdp(self):
"""
Testing how get_gdp works
"""
self.assertEqual(self.ukraine.gdp[0], 1249.4432776058,
'get_gdp does not work')
def test_investment_inflows(self):
"""
Tests how get_investment_inflows works
"""
self.assertEqual(self.bulgaria.investment_inflows[0], 0.0,
'get_investment_inflows does not work')
def test_investment_outflows(self):
"""
Tests how get_investment_outflows works
"""
self.assertEqual(self.hungary.investment_outflows[0], 2.7554412051241,
'get_investment_outflows does not work')
def test_manufacturing(self):
"""
Tests how get_manufacturing works
"""
self.assertEqual(self.poland.manufacturing[0], 19.120467211902,
'get_manufacturing does not work')
def get_estonia():
"""
gets all the data for Estonia
:return: object
"""
estonia = Country('Estonia')
estonia.get_gdp('https://www.quandl.com/api/v3/datasets/WWDI/EST'
'_NY_GDP_PCAP_CD.json?api_key=2jhCWecEKmuxzVY9ifwp')
estonia.get_investment_inflows('https://www.quandl.com/api/v3/data'
'sets/WWDI/EST_BX_KLT_DINV_WD_GD_ZS.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
estonia.get_investment_outflows('https://www.quandl.com/api/v3/dat'
'asets/WWDI/EST_BM_KLT_DINV_GD_ZS.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
estonia.get_manufacturing('https://www.quandl.com/api/v3/datasets/'
'WWDI/EST_NV_IND_MANF_ZS.json?api_key=2'
'jhCWecEKmuxzVY9ifwp')
return estonia
def get_latvia():
"""
gets all the data for Latvia
:return: object
"""
latvia = Country('Latvia')
latvia.get_gdp('https://www.quandl.com/api/v3/datasets/WWDI/LVA_NY_'
'GDP_PCAP_CD.json?api_key=2jhCWecEKmuxzVY9ifwp')
latvia.get_investment_inflows('https://www.quandl.com/api/v3/dataset'
's/WWDI/LVA_BX_KLT_DINV_WD_GD_ZS.json?'
'api_key=2jhCWecEKmuxzVY9ifwp')
latvia.get_investment_outflows('https://www.quandl.com/api/v3/data'
'sets/WWDI/LVA_BM_KLT_DINV_GD_ZS.jso'
'n?api_key=2jhCWecEKmuxzVY9ifwp')
latvia.get_manufacturing('https://www.quandl.com/api/v3/datasets'
'/WWDI/LVA_NV_IND_MANF_ZS.json?api_key='
'2jhCWecEKmuxzVY9ifwp')
return latvia
def get_croatia():
"""
get all data for Croatia
:return: object
"""
croatia = Country('Croatia')
croatia.get_gdp('https://www.quandl.com/api/v3/datasets/WWDI/HRV_N'
'Y_GDP_PCAP_CD.json?api_key=2jhCWecEKmuxzVY9ifwp')
croatia.get_investment_inflows('https://www.quandl.com/api/v3/data'
'sets/WWDI/HRV_BX_KLT_DINV_WD_GD_ZS'
'.json?api_key=2jhCWecEKmuxzVY9ifwp')
croatia.get_investment_outflows('https://www.quandl.com/api/v3/data'
'sets/WWDI/HRV_BM_KLT_DINV_GD_ZS.js'
'on?api_key=2jhCWecEKmuxzVY9ifwp')
croatia.get_manufacturing('https://www.quandl.com/api/v3/datasets/'
'WWDI/HRV_NV_IND_MANF_ZS.json?api_key='
'2jhCWecEKmuxzVY9ifwp')
return croatia
def get_slovenia():
"""
gets all the data for Slovenia
:return: object
"""
slovenia = Country('Slovenia')
slovenia.get_gdp('https://www.quandl.com/api/v3/datasets/WWDI/SVN_'
'NY_GDP_PCAP_CD.json?api_key=2jhCWecEKmuxzVY9ifwp')
slovenia.get_investment_inflows('https://www.quandl.com/api/v3/datas'
'ets/WWDI/SVN_BX_KLT_DINV_WD_GD_ZS.j'
'son?api_key=2jhCWecEKmuxzVY9ifwp')
slovenia.get_investment_outflows('https://www.quandl.com/api/v3/datas'
'ets/WWDI/SVN_BM_KLT_DINV_GD_ZS.jso'
'n?api_key=2jhCWecEKmuxzVY9ifwp')
slovenia.get_manufacturing('https://www.quandl.com/api/v3/datase'
'ts/WWDI/SVN_NV_IND_MANF_ZS.json?api_'
'key=2jhCWecEKmuxzVY9ifwp')
return slovenia
def get_czech():
"""
gets all the data for Czech Republic
:return: object
"""
czech = Country('Czech Republic')
czech.get_gdp('https://www.quandl.com/api/v3/datasets/WWDI/CZE'
'_NY_GDP_PCAP_CD.json?api_key=2jhCWecEKmuxzVY9ifwp')
czech.get_investment_inflows('https://www.quandl.com/api/v3/dat'
'asets/WWDI/CZE_BX_KLT_DINV_WD_GD_ZS'
'.json?api_key=2jhCWecEKmuxzVY9ifwp')
czech.get_investment_outflows('https://www.quandl.com/api/v3/data'
'sets/WWDI/CZE_BM_KLT_DINV_GD_ZS.js'
'on?api_key=2jhCWecEKmuxzVY9ifwp')
czech.get_manufacturing('https://www.quandl.com/api/v3/datasets'
'/WWDI/CZE_NV_IND_MANF_ZS.json?api_key='
'2jhCWecEKmuxzVY9ifwp')
return czech
def get_romania():
"""
gets all data for Romania
:return: object
"""
romania = Country('Romania')
romania.get_gdp('https://www.quandl.com/api/v3/datasets'
'/WWDI/ROU_NY_GDP_PCAP_CD.json'
'?api_key=2jhCWecEKmuxzVY9ifwp')
romania.get_investment_inflows('https://www.quandl.com/api/v3'
'/datasets/WWDI/ROU_BX_KLT_'
'DINV_WD_GD_ZS.json?api_key='
'2jhCWecEKmuxzVY9ifwp')
romania.get_investment_outflows('https://www.quandl.com/api/v3'
'/datasets/WWDI/ROU_BM_KLT_DINV'
'_GD_ZS.json?api_key=2jhCWecEK'
'muxzVY9ifwp')
romania.get_manufacturing('https://www.quandl.com/api/v3/'
'datasets/WWDI/ROU_NV_IND_MANF_'
'ZS.json?api_key=2jhCWecEKmuxzVY9ifwp')
return romania
def get_poland():
"""
gets all the data for Poland
:return: object
"""
poland = Country('Poland')
poland.get_gdp('https://www.quandl.com/api/v3'
'/datasets/WWDI/POL_NY_GDP_PCAP_CD.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
poland.get_investment_inflows('https://www.quandl.com/'
'api/v3/datasets/WWDI/'
'POL_BX_KLT_DINV_WD_GD_ZS.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
poland.get_investment_outflows('https://www.quandl.com/'
'api/v3/datasets/WWDI/'
'POL_BM_KLT_DINV_GD_ZS.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
poland.get_manufacturing('https://www.quandl.com/api'
'/v3/datasets/WWDI/POL_NV_'
'IND_MANF_ZS.json?api_key'
'=2jhCWecEKmuxzVY9ifwp')
return poland
def get_hungary():
"""
gets all the data for Hungary
:return: object
"""
hungary = Country('Hungary')
hungary.get_gdp('https://www.quandl.com/api/'
'v3/datasets/WWDI/HUN_NY_GDP_PCAP_CD.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
hungary.get_investment_inflows('https://www.quandl.com/'
'api/v3/datasets/WWDI/'
'HUN_BX_KLT_DINV_WD_GD_ZS.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
hungary.get_investment_outflows('https://www.quandl.com/'
'api/v3/datasets/WWDI/'
'HUN_BM_KLT_DINV_GD_ZS.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
hungary.get_manufacturing('https://www.quandl.com/'
'api/v3/datasets/'
'WWDI/HUN_NV_IND_MANF_ZS.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
return hungary
def get_ukraine():
"""
gets all the data for Ukraine
:return: object
"""
ukraine = Country('Ukraine')
ukraine.get_gdp('https://www.quandl.com/api/v3/'
'datasets/WWDI/UKR_NY_GDP_PCAP_CD.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
ukraine.get_investment_inflows('https://www.quandl.com'
'/api/v3/datasets/WWDI/'
'UKR_BX_KLT_DINV_WD_GD_ZS.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
ukraine.get_investment_outflows('https://www.quandl.com'
'/api/v3/datasets/WWDI/'
'UKR_BM_KLT_DINV_GD_ZS.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
ukraine.get_manufacturing('https://www.quandl.com/'
'api/v3/datasets/WWDI/'
'UKR_NV_IND_MANF_ZS.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
return ukraine
def get_bulgaria():
"""
gets all the data for Bulgaria
:return: object
"""
bulgaria = Country('Bulgaria')
bulgaria.get_gdp('https://www.quandl.com/api/v3/'
'datasets/WWDI/BGR_NY_GDP_PCAP_CD.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
bulgaria.get_investment_inflows('https://www.quandl.com'
'/api/v3/datasets/WWDI/'
'BGR_BX_KLT_DINV_WD_GD_ZS.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
bulgaria.get_investment_outflows('https://www.quandl.com/'
'api/v3/datasets/WWDI/'
'BGR_BM_KLT_DINV_GD_ZS.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
bulgaria.get_manufacturing('https://www.quandl.com/'
'api/v3/datasets/WWDI/'
'BGR_NV_IND_MANF_ZS.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
return bulgaria
from country import Country
import matplotlib.pyplot as plt
ukraine = Country('Ukraine')
ukraine.get_gdp('https://www.quandl.com/api/v3/'
'datasets/WWDI/UKR_NY_GDP_PCAP_CD.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
poland = Country('Poland')
poland.get_gdp('https://www.quandl.com/api/v3'
'/datasets/WWDI/POL_NY_GDP_PCAP_CD.'
'json?api_key=2jhCWecEKmuxzVY9ifwp')
plt.plot(ukraine.gdp)
plt.plot(poland.gdp)
plt.show()