def test_categories_to_reviews(self):
"""Tests add_categories_to_reviews to make sure it is properly running"""
category = [('categories', [CATEGORY]), ('review', TEXT)]
job = CategoryPredictor()
category_results = list(job.add_categories_to_reviews_reducer(BIZ_ID, category))
result = [('all', {CATEGORY: 1}), (CATEGORY, TEXT)]
self.assertEqual(category_results,result)
def test_categories_to_reviews(self):
"""Tests add_categories_to_reviews to make sure it is properly running"""
category = [('categories', [CATEGORY]), ('review', TEXT)]
job = CategoryPredictor()
category_results = list(
job.add_categories_to_reviews_reducer(BIZ_ID, category))
result = [('all', {CATEGORY: 1}), (CATEGORY, TEXT)]
self.assertEqual(category_results, result)
