def testGetTweetsFromScreenName(self):
"""
testing if tweets are successfully gathers from specified user with specified amount
"""
three_tweets = te.get_tweets(self.test_account, 3)
five_tweets = te.get_tweets(self.test_account, 5)
#test account only has 5 tweets
eleven_requested_tweets = te.get_tweets(self.test_account, 11)
self.assertEquals(len(three_tweets), 3,
'request for 3 tweets did not return 3 tweets')
self.assertEquals(len(five_tweets), 5,
'request for 5 tweets did not return 5 tweets')
self.assertEquals(len(eleven_requested_tweets), 8,
'request for an amount of tweets over ' +
'the accounts tweet number failed')
self.assertEquals(three_tweets[0].text, 'testacme4',
'request for 3 tweets returned invalid tweet')
self.assertEquals(three_tweets[1].text, 'testacme3',
'request for 3 tweets returned invalid tweet')
self.assertEquals(three_tweets[2].text, 'testacme2',
'request for 3 tweets returned invalid tweet')
self.assertEquals(five_tweets[0].text, 'testacme4',
'request for 5 tweets returned invalid tweet')
self.assertEquals(five_tweets[1].text, 'testacme3',
'request for 5 tweets returned invalid tweet')
self.assertEquals(five_tweets[2].text, 'testacme2',
'request for 5 tweets returned invalid tweet')
self.assertEquals(five_tweets[3].text, 'testacme1',
'request for 5 tweets returned invalid tweet')
self.assertEquals(five_tweets[4].text, 'testacme0',
'request for 5 tweets returned invalid tweet')
self.assertEquals(five_tweets[0].id, 531976729007771648,
'tweet did no have correct id')
self.assertEquals(five_tweets[1].id, 531976714730364929,
'tweet did not have correct id')
self.assertEquals(five_tweets[2].id, 531976696397070336,
'tweet did not have correct id')
self.assertEquals(five_tweets[3].id, 531976682128031744,
'tweet did not have correct id')
self.assertEquals(five_tweets[4].id, 531976660015652864,
'tweet did not have correct id')
for tweet in five_tweets:
self.assertEquals(tweet.user.screen_name, 'acmeteam4',
'tweet was not taken from the correct user')
for tweet in five_tweets:
self.assertIsInstance(tweet, tweepy.Status,
'Should be returning list of tweepy statuses, but is not')
def testStressGetTweets(self):
"""
Tests to see if twitter_explorer handles that rate call correctly, and completes task
"""
holder = []
i = 0
count = 0
while i < 3000:
holder.append(te.get_tweets('acmeteam4', 1))
i +=1# for tweets in holder:
count += 1
self.assertEquals(count, 3000, 'getting amount of tweets over rate limit failed')
def testGetSources(self):
"""
Tests to see if correct sources are gathered from specified tweets
"""
eight_tweets = te.get_tweets(self.test_account, 8)
holder = {}
for tweet in eight_tweets:
matched = te.get_sources(tweet, self.test_sources)
holder.update({tweet.id : matched})
self.assertIsInstance(matched, list,
'Not returning list of matched sources')
for element in [531976729007771648, 531976714730364929, 531976660015652864,
531976696397070336, 531976682128031744]:
self.assertEquals(holder[element],[],
'Not returning correct list of sources for each test tweet.')
#can't test exact site as using url lib will change them if they generate
#for example each google search url will be different
self.assertEquals(len(holder[531976454138241024]), 2,
'Not returning correct amount of sources for test tweet')
self.assertEquals(len(holder[531976627870527488]), 2,
'Not returning correct amount of sources for test tweet')
self.assertEquals(len(holder[531975600152805376]), 3,
'Not returning correct amount of sources for test tweet')
holder = {}
for tweet in eight_tweets:
matched = te.get_sources(tweet, [])
holder.update({tweet.id : matched})
expected = {531976714730364929: [],
531976454138241024: [],
531976660015652864: [],
531976682128031744: [],
531976729007771648: [],
531976696397070336: [],
531976627870527488: [],
531975600152805376: []}
self.assertEquals(holder, expected,
'Not returning empty list when no sources are selected.')
def testGetKeywords(self):
"""
Tests to see if correct keywords are gathered from specified tweets
"""
eight_tweets = te.get_tweets(self.test_account, 8)
holder = {}
for i in [0,1,2,3,4]:
tweet = eight_tweets[i]
matched = te.get_keywords(tweet, self.test_keywords)
holder.update({tweet.id : matched})
self.assertIsInstance(matched, list,
'Not returing list of matched keywords')
expected = {531976729007771648: ['test', 'acme4'],
531976714730364929: ['test', '3'],
531976660015652864: ['test'],
531976696397070336: ['test', '2'],
531976682128031744: ['test']}
self.assertEquals(holder, expected,
'Not returning correct list of keywords for each test tweet.')
holder = {}
for tweet in eight_tweets:
matched = te.get_keywords(tweet, [])
holder.update({tweet.id : matched})
expected = {531976714730364929: [],
531976454138241024: [],
531976660015652864: [],
531976682128031744: [],
531976729007771648: [],
531976696397070336: [],
531976627870527488: [],
531975600152805376: []}
self.assertEquals(holder, expected,
'Not returning empty list when no keywords are selected.')
