def test_two_countries(self): testGDP2A_LIE_CHE = ga.GDP2Asset() testGDP2A_LIE_CHE.set_countries(countries=['LIE', 'CHE'], path=DEMO_GDP2ASSET) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[0, 2], 9.520696799999968, 4) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[45, 2], 7.39570019999996, 4) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[1000, 2], 9.604029999999966, 4) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[2500, 2], 9.395696999999984, 4) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[0, 1], 47.062247399999976, 4) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[45, 1], 45.978915799999996, 4) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[1000, 1], 46.6039148, 4) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[2500, 1], 47.3955802, 4) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[0, 0], 174032107.65846416, 4) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[45, 0], 11682292.467251074, 4) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[1000, 0], 508470546.39168245, 4) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[2500, 0], 949321115.5175464, 4) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[0, 3], 3.0) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[12, 3], 3.0) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[0, 4], 11.0) self.assertAlmostEqual(testGDP2A_LIE_CHE.iloc[2500, 4], 11.0)
def test_one_set_countries(self): testGDP2A_LIE = ga.GDP2Asset() testGDP2A_LIE.set_countries(countries=['LIE'], path=DEMO_GDP2ASSET) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[0, 2], 9.5206968) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[1, 2], 9.5623634) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[2, 2], 9.60403) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[3, 2], 9.5206968) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[4, 2], 9.5623634) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[5, 2], 9.60403) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[6, 2], 9.5206968) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[7, 2], 9.5623634) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[8, 2], 9.60403) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[9, 2], 9.5206968) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[10, 2], 9.5623634) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[11, 2], 9.5206968) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[12, 2], 9.5623634) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[0, 1], 47.0622474) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[1, 1], 47.0622474) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[2, 1], 47.0622474) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[3, 1], 47.103914) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[4, 1], 47.103914) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[5, 1], 47.103914) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[6, 1], 47.1455806) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[7, 1], 47.1455806) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[8, 1], 47.1455806) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[9, 1], 47.1872472) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[10, 1], 47.1872472) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[11, 1], 47.2289138) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[12, 1], 47.2289138) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[0, 0], 174032107.65846416) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[1, 0], 20386409.991937194) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[2, 0], 2465206.6989314994) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[3, 0], 0.0) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[4, 0], 12003959.733058406) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[5, 0], 97119771.42771776) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[6, 0], 0.0) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[7, 0], 4137081.3646739507) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[8, 0], 27411196.308422357) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[9, 0], 0.0) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[10, 0], 4125847.312198318) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[11, 0], 88557558.43543366) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[12, 0], 191881403.05181965) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[0, 3], 3.0) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[12, 3], 3.0) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[0, 4], 11.0) self.assertAlmostEqual(testGDP2A_LIE.gdf.iloc[12, 4], 11.0)
def test_set_countries(self): testGDP2A_DEU = ga.GDP2Asset() testGDP2A_DEU.set_countries(countries=['DEU']) testGDP2A_RUS = ga.GDP2Asset() testGDP2A_RUS.set_countries(countries=['RUS']) testGDP2A_DEU_BRA = ga.GDP2Asset() testGDP2A_DEU_BRA.set_countries(countries=['DEU', 'BRA']) self.assertEqual(testGDP2A_DEU.shape[0], 26878) self.assertAlmostEqual(testGDP2A_DEU.iloc[0, 0], 706843.8067164791) self.assertAlmostEqual(testGDP2A_DEU.iloc[0, 1], 47.312247) self.assertAlmostEqual(testGDP2A_DEU.iloc[0, 2], 10.229028999999969) self.assertAlmostEqual(testGDP2A_DEU.iloc[0, 3], 3.0) self.assertAlmostEqual(testGDP2A_DEU.iloc[0, 4], 11.0) self.assertAlmostEqual(testGDP2A_DEU.iloc[26877, 0], 1054654.9189984929) self.assertAlmostEqual(testGDP2A_DEU.iloc[26877, 1], 55.0622346) self.assertAlmostEqual(testGDP2A_DEU.iloc[26877, 2], 8.437365199999988) self.assertEqual(testGDP2A_RUS.shape[0], 1698649) self.assertAlmostEqual(testGDP2A_RUS.iloc[0, 0], 316894.73009594914) self.assertAlmostEqual(testGDP2A_RUS.iloc[0, 1], 41.2289234) self.assertAlmostEqual(testGDP2A_RUS.iloc[0, 2], 47.52063599999997) self.assertAlmostEqual(testGDP2A_RUS.iloc[0, 3], 3.0) self.assertAlmostEqual(testGDP2A_RUS.iloc[0, 4], 9.0) self.assertAlmostEqual(testGDP2A_RUS.iloc[1698648, 0], 0.0) self.assertAlmostEqual(testGDP2A_RUS.iloc[1698648, 1], 81.8538584) self.assertAlmostEqual(testGDP2A_RUS.iloc[1698648, 2], 59.353950399999974) self.assertEqual(testGDP2A_DEU_BRA.shape[0], 437227) self.assertAlmostEqual(testGDP2A_DEU_BRA.iloc[0, 0], 706843.8067164791) self.assertAlmostEqual(testGDP2A_DEU_BRA.iloc[0, 1], 47.312247) self.assertAlmostEqual(testGDP2A_DEU_BRA.iloc[0, 2], 10.229028999999969) self.assertAlmostEqual(testGDP2A_DEU_BRA.iloc[0, 3], 3.0) self.assertAlmostEqual(testGDP2A_DEU_BRA.iloc[0, 4], 11.0) self.assertAlmostEqual(testGDP2A_DEU_BRA.iloc[437226, 0], 835.6131181309585) self.assertAlmostEqual(testGDP2A_DEU_BRA.iloc[437226, 1], 5.22898099999999) self.assertAlmostEqual(testGDP2A_DEU_BRA.iloc[437226, 2], -60.14585840000002) self.assertAlmostEqual(testGDP2A_DEU_BRA.iloc[437226, 3], 6.0) self.assertAlmostEqual(testGDP2A_DEU_BRA.iloc[437226, 4], 3.0)
def test_wrong_iso3_fail(self): """Wrong ISO3 code""" testGDP2A = ga.GDP2Asset() with self.assertRaises(KeyError): testGDP2A.set_countries(countries=['OYY']) with self.assertRaises(KeyError): testGDP2A.set_countries(countries=['DEU'], ref_year=2600) with self.assertRaises(ValueError): testGDP2A.set_countries()
def test_wrong_iso3_fail(self): """Wrong ISO3 code""" testGDP2A = ga.GDP2Asset() with self.assertRaises(NameError): testGDP2A.set_countries(countries=['CHE'], ref_year=2000) with self.assertRaises(NameError): testGDP2A.set_countries(countries=['CHE'], ref_year=2000, path='non/existent/test') with self.assertRaises(LookupError): testGDP2A.set_countries(countries=['OYY'], path=DEMO_GDP2ASSET) with self.assertRaises(LookupError): testGDP2A.set_countries(countries=['DEU'], ref_year=2600, path=DEMO_GDP2ASSET) with self.assertRaises(ValueError): testGDP2A.set_countries(path=DEMO_GDP2ASSET)
def test_wrong_iso3_fail(self): """Wrong ISO3 code""" testGDP2A = ga.GDP2Asset() with self.assertRaises(KeyError): testGDP2A.set_countries(countries=['OYY'], path=DEMO_GDP2ASSET) with self.assertRaises(KeyError): testGDP2A.set_countries(countries=['DEU'], ref_year=2600, path=DEMO_GDP2ASSET) with self.assertRaises(KeyError): testGDP2A.set_countries(countries=['DEU'], ref_year=2600, path=DEMO_GDP2ASSET) with self.assertRaises(ValueError): testGDP2A.set_countries(path=DEMO_GDP2ASSET) with self.assertRaises(IOError): testGDP2A.set_countries(countries=['MEX'], path=DEMO_GDP2ASSET)