def main(): user = input("User handle to analyze?\n") word_list = tweets_to_list(get_all_tweets(user)) sentences = 0 chain = MarkovChain(word_list) while True: try: sentences = int(input("How many sentences to generate?\n")) break except ValueError: print("Please input a number\n") for _ in range(sentences): chain.generate_sentence()
def test_generate_sentence(self): mc = MarkovChain() mc.add(('aaa', 'bbb.', 'ccc')) self.assertEqual('Aaa bbb.', mc.generate_sentence(12)) self.assertEqual('Aaa bbb. Ccc.', mc.generate_sentence(13))
def r(): mv = MarkovChain(int(request.form.get('n'))) mv.build_markov('data/sherlock_holmes.txt') return ' '.join(mv.generate_sentence())