def test_plot(self): '''This function will test if plots from year 2007 to 2012 are generated successfully''' try: for i in range(2007,2013): os.remove('./Income_Box_'+str(i)+'.jpg') os.remove('./Income_Hist_'+str(i)+'.jpg') except OSError: pass i = Income() i.plot_2007_2012() for i in range(2007,2013): self.assertTrue(os.path.isfile('./Income_Box_'+str(i)+'.jpg'), "Fail to generate plot.") self.assertTrue(os.path.isfile('./Income_Hist_'+str(i)+'.jpg'), "Fail to generate plot.")
def main(): income_data = Income() # Ask user for a year and exit until 'finish' entered. user_input = get_user_input() while user_input != "finish": year = 0 while year == 0: try: year = read_year(user_input) income_data.income_dist(year) break except ValueError: print "Invalid input. Please enter again." except RangeError: print "The year you entered is not in [1800, 2012]. Please enter again." user_input = get_user_input() user_input = get_user_input() # plot income data from 2007 to 2015 income_data.plot_2007_2012()