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 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)