def main():
# Q7
import_and_transform('../indicator gapminder gdp_per_capita_ppp.xlsx')
try:
while True:
print '\n You have three options'
print '\n 1. Please input a year between 1800 and 2012 to generate plots'
print '\n 2. Please enter finish to exit plotting and generate graphs for the years 2007-2012'
print '\n 3. Please enter quit to exit this program\n'
x = raw_input()
if x == 'finish':
break
elif x == 'quit':
sys.exit()
elif re.match(r"^[0-9]{4}$", x):
try:
plot_income_by_yr(int(x))
except KeyError:
print '\n Input Error: Invalid Year!\n'
else:
print "\n Input Error: Please a valid year!\n"
# Q8
print '\n Generating plots......'
yr_list = range(2007, 2013)
for yr in yr_list:
result = vs.visualize_tool(yr) # use visulize tool class to generate plot
result.plot_boxplot()
result.plot_hist()
print '\n Plots are saved at current dictionary'
except KeyboardInterrupt, ValueError:
print "\n Interrupted!"
def test_vs_2(self):
with self.assertRaises(KeyError) as cm:
vs.visualize_tool(9000)
the_exception = cm.exception
self.assertEquals(str(the_exception), "'Invalid Year!'")