def testMerge(self):
self.assertEqual(merge_by_year(1820, income, countries).shape[0], 3)
def testMerge(self):
self.assertEqual(merge_by_year(1820, income, countries).shape[0], 3)
cwd = os.getcwd()
countries = pd.read_csv(cwd + '/countries.csv')
income = pd.read_excel(cwd + '/indicator gapminder gdp_per_capita_ppp.xlsx',
index_col = 0).T
print(income.head())
year = 0
while year != 'finish':
year = input('Enter a year to see income visualization. Enter \'finish\' to quit. ')
try:
year = int(year)
if year not in income.index:
print('Not valid year.')
else:
plot_year(year, income)
except ValueError:
if year == 'finish':
pass
else:
print('Not valid year.')
for year in range(2007, 2013):
df_year= merge_by_year(year, income, countries)
Region(year, df_year)
if __name__ == '__main__':
pass
