def test_tweet_is_correct():
judger = Judger()
# dummy blacklist class injection
judger.blacklists = DummyBlacklists()
for trend in sample_trends:
should_tweet = judger.judge_whether_tweet(trend)
if trend.get('name') == 'trend_A':
assert should_tweet is False
elif trend.get('name') == 'trend_B':
assert should_tweet is False
elif trend.get('name') == 'trend_C':
assert should_tweet is True
elif trend.get('name') == 'trend_D':
assert should_tweet is True
elif trend.get('name') == 'trend_E':
assert should_tweet is True
elif trend.get('name') == 'trend_F':
assert should_tweet is False
elif trend.get('name') == 'trend_blacklisted_A':
assert should_tweet is False
def test_tweet_volume_lager_than_threshold():
trend = {'name': 'dummy', 'tweet_volume': THRETHOLD + 1}
judger = Judger()
judger.blacklists = DummyBlacklists()
assert judger.judge_whether_tweet(trend)
def test_tweet_volume_equals_threshold():
trend = {'name': 'dummy', 'tweet_volume': THRETHOLD}
judger = Judger()
judger.blacklists = DummyBlacklists()
assert not judger.judge_whether_tweet(trend)
del judger
