class TwitterTrainerTestCase(ChatBotTestCase):
def setUp(self):
"""
Instantiate the trainer class for testing.
"""
super(TwitterTrainerTestCase, self).setUp()
self.trainer = TwitterTrainer(
self.chatbot,
twitter_consumer_key='twitter_consumer_key',
twitter_consumer_secret='twitter_consumer_secret',
twitter_access_token_key='twitter_access_token_key',
twitter_access_token_secret='twitter_access_token_secret',
show_training_progress=False)
self.trainer.api = Mock()
self.trainer.api.GetSearch = MagicMock(
side_effect=get_search_side_effect)
self.trainer.api.GetStatus = MagicMock(
side_effect=get_status_side_effect)
def test_random_word(self):
word = self.trainer.random_word('random')
self.assertTrue(len(word) > 3)
def test_get_words_from_tweets(self):
tweets = get_search_side_effect()
words = self.trainer.get_words_from_tweets(tweets)
self.assertIn('about', words)
self.assertIn('favorite', words)
self.assertIn('things', words)
def test_get_statements(self):
statements = self.trainer.get_statements()
self.assertEqual(len(statements), 1)
def test_train(self):
self.trainer.train()
statements = list(self.trainer.chatbot.storage.filter())
self.assertGreater(len(statements), 1)
def test_train_sets_search_text(self):
self.trainer.train()
statements = list(self.trainer.chatbot.storage.filter())
self.assertGreater(len(statements), 1)
self.assertEqual(statements[0].search_text, 'PRP:sure IN:jewel')
def test_train_sets_search_in_response_to(self):
self.trainer.train()
statements = list(self.trainer.chatbot.storage.filter())
self.assertGreater(len(statements), 1)
self.assertEqual(statements[0].search_in_response_to,
'PRP:sure IN:jewel')
class TwitterTrainerTestCase(ChatBotTestCase):
def setUp(self):
"""
Instantiate the trainer class for testing.
"""
super(TwitterTrainerTestCase, self).setUp()
self.trainer = TwitterTrainer(
self.chatbot,
twitter_consumer_key='twitter_consumer_key',
twitter_consumer_secret='twitter_consumer_secret',
twitter_access_token_key='twitter_access_token_key',
twitter_access_token_secret='twitter_access_token_secret',
show_training_progress=False
)
self.trainer.api = Mock()
self.trainer.api.GetSearch = MagicMock(side_effect=get_search_side_effect)
self.trainer.api.GetStatus = MagicMock(side_effect=get_status_side_effect)
def test_random_word(self):
word = self.trainer.random_word('random')
self.assertTrue(len(word) > 3)
def test_get_words_from_tweets(self):
tweets = get_search_side_effect()
words = self.trainer.get_words_from_tweets(tweets)
self.assertIn('about', words)
self.assertIn('favorite', words)
self.assertIn('things', words)
def test_get_statements(self):
statements = self.trainer.get_statements()
self.assertEqual(len(statements), 1)
def test_train(self):
self.trainer.train()
statements = list(self.trainer.chatbot.storage.filter())
self.assertGreater(len(statements), 1)
def test_train_sets_search_text(self):
self.trainer.train()
statements = list(self.trainer.chatbot.storage.filter())
self.assertGreater(len(statements), 1)
self.assertEqual(statements[0].search_text, 'urub')
def test_train_sets_search_in_response_to(self):
self.trainer.train()
statements = list(self.trainer.chatbot.storage.filter())
self.assertGreater(len(statements), 1)
self.assertEqual(statements[0].search_in_response_to, 'urub')
class TwitterTrainerTestCase(ChatBotTestCase):
def setUp(self):
"""
Instantiate the trainer class for testing.
"""
super(TwitterTrainerTestCase, self).setUp()
self.trainer = TwitterTrainer(
self.chatbot.storage,
twitter_consumer_key='twitter_consumer_key',
twitter_consumer_secret='twitter_consumer_secret',
twitter_access_token_key='twitter_access_token_key',
twitter_access_token_secret='twitter_access_token_secret')
self.trainer.api = Mock()
self.trainer.api.GetSearch = MagicMock(
side_effect=get_search_side_effect)
self.trainer.api.GetStatus = MagicMock(
side_effect=get_status_side_effect)
def test_random_word(self):
word = self.trainer.random_word('random')
self.assertTrue(len(word) > 3)
def test_get_words_from_tweets(self):
tweets = get_search_side_effect()
words = self.trainer.get_words_from_tweets(tweets)
self.assertIn('about', words)
self.assertIn('favorite', words)
self.assertIn('things', words)
def test_get_statements(self):
statements = self.trainer.get_statements()
self.assertEqual(len(statements), 1)
def test_train(self):
self.trainer.train()
statement_created = self.trainer.storage.filter()
self.assertTrue(len(statement_created))
