def test_dig_for_twins(self):
MockTweet = namedtuple('MockTweet', ['text', 'user', 'id'])
MockUser = namedtuple('MockUser', ['screen_name'])
tweets = [
MockTweet('Hello world', MockUser('user1'), 1),
MockTweet('Hello world', MockUser('user2'), 2)]
res = dig_for_twins(tweets)
self.assertEqual(len(res), 2)
def twinsy_finder(fetch_size=5000):
print("Running twinsy finder...")
fetched_tweets = fetch_tweets('Kanye', fetch_size=fetch_size)
tweets = dig_for_twins(fetched_tweets)
if tweets:
print("Twins found, updating status.")
update_status(tweets)
else:
print("No twins found.")
def test_dig_for_twins_repeat_user(self):
MockTweet = namedtuple('MockTweet', ['text', 'user', 'id'])
MockUser = namedtuple('MockUser', ['screen_name'])
tweets = [
MockTweet('Hello world', MockUser('user1'), 1),
MockTweet('Hello world', MockUser('user1'), 2),
MockTweet('Hello world', MockUser('user1'), 3)]
res = dig_for_twins(tweets)
self.assertIsNone(res)
def twinsy_finder(fetch_size=5000):
print("Running twinsy finder...")
fetched_tweets = fetch_tweets('Kanye', fetch_size=fetch_size)
tweets = dig_for_twins(fetched_tweets)
if tweets:
print("Twins found, updating status.")
update_status(tweets)
else:
print("No twins found.")
def test_dig_for_twins_miss(self):
MockTweet = namedtuple('MockTweet', ['text', 'user', 'id'])
MockUser = namedtuple('MockUser', ['screen_name'])
tweets = [
MockTweet('Hello world', MockUser('user1'), 1),
MockTweet('Different tweet', MockUser('user2'), 2)
]
res = dig_for_twins(tweets)
self.assertIsNone(res)
def test_dig_for_twins_contacted(self):
CONTACTED_SCREEN_NAMES.add('user1')
CONTACTED_SCREEN_NAMES.add('user2')
CONTACTED_SCREEN_NAMES.add('user3')
MockTweet = namedtuple('MockTweet', ['text', 'user', 'id'])
MockUser = namedtuple('MockUser', ['screen_name'])
tweets = [
MockTweet('Hello world', MockUser('user1'), 1),
MockTweet('Hello world', MockUser('user2'), 2),
MockTweet('Hello world', MockUser('user3'), 3)]
res = dig_for_twins(tweets)
self.assertIsNone(res)
def test_dig_for_twins_contacted(self):
CONTACTED_SCREEN_NAMES.add('user1')
CONTACTED_SCREEN_NAMES.add('user2')
CONTACTED_SCREEN_NAMES.add('user3')
MockTweet = namedtuple('MockTweet', ['text', 'user', 'id'])
MockUser = namedtuple('MockUser', ['screen_name'])
tweets = [
MockTweet('Hello world', MockUser('user1'), 1),
MockTweet('Hello world', MockUser('user2'), 2),
MockTweet('Hello world', MockUser('user3'), 3)
]
res = dig_for_twins(tweets)
self.assertIsNone(res)
