def test_share_year(self): Analyzer = CompanyShare("valid_data.csv") self.result=Analyzer.getShareResult() self.assertEquals(1990, self.result[6].year) print '\n Results \n' #print self.result[6].year Analyzer.printResults()
def test_invalid_data(self): #TEST FOR Invalid Data FILE Analyzer = CompanyShare("invalid_data.csv") self.assertRaises(InvalidException, lambda: Analyzer.getShareResult())