def test_consider_retweets_protected_does_not_retweet(self):
self.faux_tweets[0]['protected'] = True
self.mock_CLS.classify.return_value = [[0, 1]]
brain._consider_retweets(self.faux_tweets)
self.assertFalse(self.mock_twitter.retweet.called)
def test_consider_retweets_below_threshold_does_not_retweet(self):
self.mock_CLS.classify.return_value = [[1, 0]]
brain._consider_retweets(self.faux_tweets)
self.assertFalse(self.mock_twitter.retweet.called)
def test_consider_retweets_max_retweets_does_not_retweet(self):
self.faux_config.max_retweets = 0
self.mock_CLS.classify.return_value = [[0, 1]]
brain._consider_retweets(self.faux_tweets)
self.assertFalse(self.mock_twitter.retweet.called)
def test_consider_retweets_above_threshold_does_retweet(self):
self.mock_CLS.classify.return_value = [[0, 1]]
brain._consider_retweets(self.faux_tweets)
self.mock_twitter.retweet.assert_called_with(1234)
def test_consider_retweets_protected_does_not_retweet(self):
self.faux_tweets[0]['protected'] = True
self.mock_CLS.classify.return_value = [[0,1]]
brain._consider_retweets(self.faux_tweets)
self.assertFalse(self.mock_twitter.retweet.called)
def test_consider_retweets_max_retweets_does_not_retweet(self):
self.faux_config.max_retweets = 0
self.mock_CLS.classify.return_value = [[0,1]]
brain._consider_retweets(self.faux_tweets)
self.assertFalse(self.mock_twitter.retweet.called)
def test_consider_retweets_below_threshold_does_not_retweet(self):
self.mock_CLS.classify.return_value = [[1,0]]
brain._consider_retweets(self.faux_tweets)
self.assertFalse(self.mock_twitter.retweet.called)
def test_consider_retweets_above_threshold_does_retweet(self):
self.mock_CLS.classify.return_value = [[0,1]]
brain._consider_retweets(self.faux_tweets)
self.mock_twitter.retweet.assert_called_with(1234)