def test_get_keyword(self): expected = ['Asylum Games', 'Banjooli', 'Mutinies', 'Polis'] rm = ReportModel(Reports.asylum) keywords = rm.getKeywords() #print keywords self.assertEqual(len(expected), len(keywords)) self.assertEqual(expected, keywords)
def test_get_keyword_dont_return_special_keys(self): expected = ['Polis', 'Banjooli', 'Mutinies', 'Asylum Games'] report = {'Asylum Games':[], 'Banjooli':[], 'Mutinies':[], 'Polis':[], 'title':"", 'report_date':""} rm = ReportModel(report) keywords = rm.getKeywords() print keywords self.assertEqual(len(expected), len(keywords)) self.assertEqual(expected, keywords)