def test_review_category(self):
"""Test the review_category_mapper function with a mock input"""
review = REVIEW_TEMPLATE % (TEXT, BIZ_NAME)
business = BUSINESS_TEMPLATE % (CATEGORY, BIZ_NAME)
job = WeightedPositiveWords()
review_results = list(job.review_category_mapper(None, json.loads(review)))
biz_results = list(job.review_category_mapper(None, json.loads(business)))
review_after_results = [(BIZ_NAME, ('review', (TEXT, 3)))]
biz_after_results = [(BIZ_NAME, ('categories', [CATEGORY]))]
self.assertEqual(review_results, review_after_results)
self.assertEqual(biz_results, biz_after_results)
