Example #1
0
def test_return_tweets_from_user():
    username = '******'
    search_tweets_task = st.SearchTweetsTask(from_username=username,
                                             tweets_limit=100)
    tweets_collector = st.CollectorTweetOutput()
    st.TweetSearchRunner(search_tweets_task=search_tweets_task,
                         tweet_outputs=[tweets_collector]).run()
    tweet_list_assert_condition(tweets_collector.get_scrapped_tweets(),
                                lambda tweet: tweet.user_name == username)
Example #2
0
def test_exact_words():
    exact_phrase = 'duda kaczyƄski kempa'
    search_tweets_task = st.SearchTweetsTask(exact_words=exact_phrase)
    tweets_collector = st.CollectorTweetOutput()
    st.TweetSearchRunner(search_tweets_task=search_tweets_task,
                         tweet_outputs=[tweets_collector]).run()
    tweet_list_assert_condition(
        tweets_collector.get_raw_list(), lambda tweet: to_base_text(
            exact_phrase) in to_base_text(tweet.full_text))
Example #3
0
def _run_test_between_dates(since: Arrow, until: Arrow):
    search_tweets_task = st.SearchTweetsTask(any_word="#koronawirus #covid19",
                                             since=since,
                                             until=until)
    tweets_collector = st.CollectorTweetOutput()
    st.TweetSearchRunner(search_tweets_task=search_tweets_task,
                         tweet_outputs=[tweets_collector]).run()
    tweet_list_assert_condition(
        tweets_collector.get_scrapped_tweets(),
        lambda tweet: since <= tweet.created_at <= until)
Example #4
0
def test_search_to_username():
    username = '******'
    search_tweets_task = st.SearchTweetsTask(to_username=username,
                                             tweets_limit=100)
    tweets_collector = st.CollectorTweetOutput()
    st.TweetSearchRunner(search_tweets_task=search_tweets_task,
                         tweet_outputs=[tweets_collector]).run()
    tweet_list_assert_condition(
        tweets_collector.get_scrapped_tweets(),
        lambda tweet: to_base_text(username) in to_base_text(tweet.full_text))
def _run_search_test_covid_tweets_in_language(language: st.Language):
    search_tweets_task = st.SearchTweetsTask(all_words='#covid19',
                                             tweets_limit=100,
                                             language=language)
    tweets_collector = st.CollectorTweetOutput()
    st.TweetSearchRunner(search_tweets_task=search_tweets_task,
                         tweet_outputs=[tweets_collector]).run()
    tweet_list_assert_condition(
        tweets_collector.get_scrapped_tweets(),
        lambda tweet: tweet.lang == language.short_value)
Example #6
0
def run_test_for_single_language(language: st.Language):
    search_tweets_task = st.SearchTweetsTask(all_words='#covid19',
                                             tweets_limit=10,
                                             language=language)
    tweets_collector = st.CollectorTweetOutput()
    st.TweetSearchRunner(search_tweets_task=search_tweets_task,
                         tweet_outputs=[tweets_collector]).run()
    tweet_list_assert_condition(
        tweets_collector.get_raw_list(),
        lambda tweet: tweet.lang in language.short_value)
Example #7
0
def test_any_word():
    any_phrase = 'kaczynski tusk'
    search_tweets_task = st.SearchTweetsTask(any_word=any_phrase,
                                             tweets_limit=100)
    tweets_collector = st.CollectorTweetOutput()
    st.TweetSearchRunner(search_tweets_task=search_tweets_task,
                         tweet_outputs=[tweets_collector]).run()

    tweet_list_assert_condition(
        tweets_collector.get_raw_list(),
        lambda tweet: contains_any_word(any_phrase, tweet.full_text) or
        contains_any_word(any_phrase, tweet.user_full_name
                          ) or contains_any_word(any_phrase, tweet.user_name))
Example #8
0
def test_search_as_replay():
    search_tweets_task = st.SearchTweetsTask(
        all_words='#covid19',
        tweets_limit=500,
        replies_filter=st.RepliesFilter.ONLY_REPLIES
    )
    tweets_collector = st.CollectorTweetOutput()
    st.TweetSearchRunner(
        search_tweets_task=search_tweets_task,
        tweet_outputs=[tweets_collector]
    ).run()
    tweet_list_assert_condition(
        tweets_collector.get_raw_list(),
        lambda tweet: len(tweet.in_reply_to_status_id_str + tweet.in_reply_to_user_id_str) > 0
    )