def test_check_get_shift_codes_returns_only_good_shift_code_from_text_with_1_good_shift_code_and_2_bad_shift_codes(
self):
with open(GOOD_1_SHIFT_CODE_2_BAD_SHIFT_CODES_TWEET_FILE) as f:
tweet = f.read()
actual = tweet_analyzer.get_shift_codes(tweet)
self.assertEqual(1, len(actual))
self.assertEqual(GOOD_SHIFT_CODE, actual[0])
def test_check_get_shift_codes_returns_2_shift_codes_from_tweet_with_2_shift_codes_in_different_places(
self):
with open(GOOD_2_SHIFT_CODES_IN_DIFFERENT_PLACES_TWEET_FILE) as f:
tweet = f.read()
actual = tweet_analyzer.get_shift_codes(tweet)
self.assertEqual(2, len(actual))
self.assertEqual(GOOD_SHIFT_CODE, actual[0])
self.assertEqual(GOOD_SHIFT_CODE_2, actual[1])
def test_check_get_shift_codes_returns_empty_list_when_tweet_does_not_contain_shift_code(
self):
with open(TWEET_FILE_WITHOUT_SHIFT_CODE) as f:
tweet = f.read()
actual = tweet_analyzer.get_shift_codes(tweet)
self.assertEqual(0, len(actual))
def test_check_get_shift_codes_returns_shift_code_from_tweet(self):
with open(GOOD_SHIFT_CODE_TWEET_FILE) as f:
tweet = f.read()
actual = tweet_analyzer.get_shift_codes(tweet)
self.assertEqual(1, len(actual))
self.assertEqual(GOOD_SHIFT_CODE, actual[0])