def runTestsAndPrintReport(trainingCategories, testingCategories): allPositiveTrainingReviews = list(); allNegativeTrainingReviews = list(); allPositiveTestingReviews = list(); allNegativeTestingReviews = list(); for category in trainingCategories: (allPositiveTrainingReviewsTmp, allNegativeTrainingReviewsTmp) = getAllClassifiedReviews(AmazonReviewCorpusReader(), category) allPositiveTrainingReviews += allPositiveTrainingReviewsTmp; allNegativeTrainingReviews += allNegativeTrainingReviewsTmp; for category in testingCategories: (allPositiveTestingReviewsTmp, allNegativeTestingReviewsTmp) = getAllClassifiedReviews(AmazonReviewCorpusReader(), category) allPositiveTestingReviews += allPositiveTestingReviewsTmp; allNegativeTestingReviews += allNegativeTestingReviewsTmp; datasets = [ClassifiedDataset(0.1, 'POS', allPositiveTrainingReviews, allPositiveTestingReviews), ClassifiedDataset(0.1, 'NEG', allNegativeTrainingReviews, allNegativeTestingReviews) ] print 'training: %s vs testing: %s' % (trainingCategories, testingCategories); report = Report(datasets, factories) reportStats = report.getReport(); Report.printReport(reportStats);
def testGetReport(self): rep = Report() self.assertTrue(rep.saveReport("Escola Santa Clara","Juan","professor",0,1,0,1,0,2,1,0,0,1)) form = {} form['qualification'] = 'professor' form['teacherName'] = 'Juan' form['question0'] = 0 form['question1'] = 1 form['question2'] = 0 form['question3'] = 1 form['question4'] = 0 form['question5'] = 2 form['question6'] = 1 form['question7'] = 0 form['question8'] = 0 form['question9'] = 1 self.assertEqual(rep.getReport('Escola Santa Clara'),form)