Exemple #1
0
	def test_smoke(self):
		"""Does a full run of weighted positive words"""

		# Need 3 mock businesses to test
		business1 = BUSINESS_TEMPLATE % (CATEGORY, "Yelp")
		business2 = BUSINESS_TEMPLATE % (CATEGORY, "Target")
		business3 = BUSINESS_TEMPLATE % (CATEGORY, "Walmart") 
		# Need more than 1 review for weighted threshold
		review1 = REVIEW_TEMPLATE % (TEXT, "Yelp")
		review2 = REVIEW_TEMPLATE % (TEXT, "Target")
		review3 = REVIEW_TEMPLATE % (TEXT, "Walmart")

		# Need at least 50 occurrences of reviews, so multiply the first review by 20
		total_input = (business1 + business2 + business3
			+ (review1 * 20) + review2 + review3)
		static_stdin = StringIO(total_input)

		job = WeightedPositiveWords(['-r', 'inline', '--no-conf', '-'])
		job.sandbox(stdin=static_stdin)

		results = []
		with job.make_runner() as runner:
			runner.run()
			for line in runner.stream_output():
				key, value = job.parse_output_line(line)
				results.append(value)
		end_result = [[CATEGORY, 66.0, 'hello'], [CATEGORY, 66.0, 'world']]
		self.assertEqual(results, end_result)