def test_classify(hashtags, full_text, expected_classification):
tweet = Tweet(hashtags=hashtags, full_text=full_text)
tweet_classifier = TweetClassifier(
tweet, classification_model=test_classification_model)
tweet_classifier.classify()
assert expected_classification == tweet.classification
def test_weights(tag_count, not_tag_count, full_text, hashtags, weight_text,
weight_hashtag):
expected_classification = {}
if tag_count:
expected_classification['tag'] = tag_count
if not_tag_count:
expected_classification['not_tag'] = not_tag_count
tweet = Tweet(full_text=full_text, hashtags=hashtags)
tweet_classifier = TweetClassifier(
tweet,
classification_model=test_classification_model,
weight_text=weight_text,
weight_hashtag=weight_hashtag)
hashtag_classification = tweet_classifier._classify_hashtags()
full_text_classification = tweet_classifier._classify_full_text()
classification = hashtag_classification
classification += full_text_classification
assert expected_classification == classification
def test_classify_full_text(mock_classify_words):
mock_tweet = MagicMock()
tweet_classifier = TweetClassifier(
mock_tweet, classification_model=test_classification_model)
classified_full_text = tweet_classifier._classify_full_text()
assert mock_classify_words.return_value == classified_full_text
mock_classify_words.assert_called_with(mock_tweet.full_text, 1)
def test_classify_hashtags(mock_classify_words):
mock_tweet = MagicMock()
tweet_classifier = TweetClassifier(
mock_tweet, classification_model=test_classification_model)
classified_hashtags = tweet_classifier._classify_hashtags()
assert mock_classify_words.return_value == classified_hashtags
mock_classify_words.assert_called_with(mock_tweet.hashtags,
3,
delimiter='|')
def test_tweet_classifier_init():
tweet = Tweet()
tweet_classifier = TweetClassifier(tweet)
assert tweet == tweet_classifier.tweet
assert global_classification_model == tweet_classifier.classification_model
tweet_classifier = TweetClassifier(
tweet, classification_model=test_classification_model)
assert tweet == tweet_classifier.tweet
assert test_classification_model == tweet_classifier.classification_model
def test_classify_words(tag_count, not_tag_count, words, delimiter):
expected_classification = {}
if tag_count:
expected_classification['tag'] = tag_count
if not_tag_count:
expected_classification['not_tag'] = not_tag_count
tweet_classifier = TweetClassifier(
Tweet(), classification_model=test_classification_model)
classification = tweet_classifier._classify_words(words,
weight=1,
delimiter=delimiter)
assert expected_classification == classification
def classify_tweets(self):
[
TweetClassifier(
tweet,
classification_model=self.classification_model).classify()
for tweet in self.tweets
]