def test_process_raw_tweets_success(self):
#Using previously pickled data of 5 tweets:
raw_tweets = pickle.load(open("test/raw_tweets.pickle", "rb"))
twemail = Twemail()
processed_tweets = twemail._process_raw_tweets(raw_tweets)
self.assertEqual(len(processed_tweets), 5)
def test_process_raw_tweets_fail(self):
#this data is an unsuccessful (unauthenticated) attempt
raw_tweets = pickle.load(open("test/raw_tweets_fail.pickle", "rb"))
twemail = Twemail()
processed_tweets = twemail._process_raw_tweets(raw_tweets)
self.assertEqual(len(processed_tweets), 0)
def test_media_extraction(self):
raw_tweets = pickle.load(open("test/tweet_with_image.pickle", "rb"))
twemail = Twemail()
parsed_tweet = twemail._process_raw_tweets(raw_tweets)[0]
self.assertEqual(parsed_tweet["links"], [
{"start":122, "end":144, "url":"http://pbs.twimg.com/media/ByopT3_CQAAZkyU.jpg"}]) #image
