class TweepyResultSetTests(unittest.TestCase): def setUp(self): self.results = ResultSet() for i in ids_fixture: self.results.append(IdItem(i)) self.results.append(NoIdItem()) def testids(self): ids = self.results.ids() self.assertListEqual(ids, ids_fixture) def testmaxid(self): self.assertEqual(self.results.max_id, 100) def testsinceid(self): self.assertEqual(self.results.since_id, 1)
def extract_users_tweets(handle, number): """ Argument Order: handle, number of tweets to extract Extract's a user's tweets """ final = ResultSet() #can change to resultset later if I want try: for status in tweepy.Cursor(api.user_timeline, screen_name=handle, count=200, include_rts=True).items(number): final.append(status) except: print("{} is a protected user!") return [] return final
def pages(cls, _: int) -> Iterator[ResultSet]: json_name = 'timeline' (_, kwargs) = mock_cursor.call_args if 'since_id' not in kwargs: json_name = 'fav' json_path = f'{JSON_DIR}/twitter/tweets/{json_name}.json' tweets: list = load_json(json_path) tweet_list: ResultSet = ResultSet(max_id=1, since_id=1) for tweet_dict in tweets: tweet_list.append(tweepy.Status.parse(tweepy.api, tweet_dict)) return iter([tweet_list])
def setUp(self): self.results = ResultSet() for i in ids_fixture: self.results.append(IdItem(i)) self.results.append(NoIdItem())