Fixed some tests so they are more consistent overall

positive_category_words/test_weighted_positivity.py

@@ -10,7 +10,7 @@
 COMPANY = u'Company'
 REVIEW_TEMPLATE = '{"type":"review", "stars":3, "text":"%s",\
 "business_id":"%s"}\n'
-BUSINESS_TEMPLATE = '{"type":"business", "categories":["%s"], \
+BUSINESS_TEMPLATE = '{"type":"business", "categories":["Company"], \
 "business_id":"%s"}\n'
 TEXT = u"Hello world"
 BIZ_NAME = u'Qdoba'
@@ -55,8 +55,8 @@ def test_review_category(self):
 		business = BUSINESS_TEMPLATE % BIZ_NAME
 
 		job = WeightedPositiveWords()
-		results = (BIZ_NAME, ('review', (TEXT, 3))),
-		(BIZ_NAME, ('categories', [COMPANY]))
+		results = ((BIZ_NAME, ('review', (TEXT, 3))),
+		(BIZ_NAME, ('categories', [COMPANY])))
 		self.assertEqual(
 			job.review_category_mapper(None, json.loads(review)).next(),
 			results[0]
@@ -87,7 +87,7 @@ def test_review_mapper(self):
 		biz_review_positivity = (BIZ_NAME, (TEXT, 3))
 
 		job = WeightedPositiveWords()
-		results = ((COMPANY, 'awesome'), (BIZ_NAME, 3))
+		results = ((COMPANY, u'world'), (BIZ_NAME, 3))
 		self.assertEqual(
 			job.review_mapper(COMPANY, biz_review_positivity).next(),
 			results

review_autopilot/test_autopilot.py

@@ -37,9 +37,8 @@ def test_smoke(self):
 		too long."""
 		#Random data to feed into the markov model.
 		#I use long runs of foo to get through the threshold filters.
-		text = '"foo bar foo baz foo car foo daz' + 'foo' * 10 + 'foofoo yelp \
-foo yar foo foo bar bar dar"'
-
+		text = 'foo bar foo baz foo car foo daz ' + ('foo ' * 10) + 'foofoo yelp \
+foo yar foo foo bar bar dar'
 		single_review = REVIEW_TEMPLATE % (text, BIZ)
 		business = BUSINESS_TEMPLATE % (CATEGORIES, BIZ)
 		static_stdin = StringIO(single_review + business)
@@ -70,7 +69,7 @@ def test_categories_reducer(self):
 	def test_split_mapper(self):
 		"""Tests split_mapper reducer in autopilot"""
 		job = ReviewAutoPilot()
-		TEST_RETURN = (('is', 'F'), ('a', 1))
+		TEST_RETURN = (('hello', 'C'), ('<end>', 1))
 		self.assertEqual(job.review_split_mapper(CATEGORIES, TEXT).next(),
 			TEST_RETURN)