def test_review_category(self):
"""Tests the category_mapper to make sure it is properly running"""
business = BUSINESS_TEMPLATE % (CATEGORY, BIZ_ID)
review = REVIEW_TEMPLATE % (TEXT, BIZ_ID)
job = CategoryPredictor()
review_results = list(job.review_category_mapper(None, json.loads(review)))
biz_results = list(job.review_category_mapper(None, json.loads(business)))
self.assertEqual(review_results, [(BIZ_ID, ('review', TEXT))])
self.assertEqual(biz_results, [(BIZ_ID, ('categories', [CATEGORY]))])
def test_review_category(self):
"""Tests the category_mapper to make sure it is properly running"""
business = BUSINESS_TEMPLATE % (CATEGORY, BIZ_ID)
review = REVIEW_TEMPLATE % (TEXT, BIZ_ID)
job = CategoryPredictor()
review_results = list(
job.review_category_mapper(None, json.loads(review)))
biz_results = list(
job.review_category_mapper(None, json.loads(business)))
self.assertEqual(review_results, [(BIZ_ID, ('review', TEXT))])
self.assertEqual(biz_results, [(BIZ_ID, ('categories', [CATEGORY]))])
