def test_tokenize_reviews(self):
"""Tests tokenize_reviews_mapper to make sure it is properly running"""
review = {CATEGORY: 1}
job = CategoryPredictor()
token_results = list(job.tokenize_reviews_mapper('all', review))
result = [('all', {CATEGORY: 1})]
self.assertEqual(token_results, result)
def test_tokenize_reviews(self):
"""Tests tokenize_reviews_mapper to make sure it is properly running"""
review = {CATEGORY: 1}
job = CategoryPredictor()
token_results = list(job.tokenize_reviews_mapper('all', review))
result = [('all', {CATEGORY: 1})]
self.assertEqual(token_results, result)
