def test_append_word_with_go_to_list(self): keyword_generator = KeywordGenerator() test_list = ['test', 'ok'] self.assertEqual( ['test', 'ok', 'gotest', 'gook'], keyword_generator.append_word_with_go_to_list(test_list)) # No words test_list = [] self.assertEqual( [], keyword_generator.append_word_with_go_to_list(test_list))
def test_generate_search_terms(self): keyword_generator = KeywordGenerator() # Fake ID self.assertEqual( [], keyword_generator.generate_search_terms('FAKE TEAM ID')) # Actual ID expected_list = [ 'TrueToAtlanta', 'TrueToAtlanta', 'ATL', 'Hawks', 'Atlanta', 'goTrueToAtlanta', 'goTrueToAtlanta', 'goATL', 'goHawks', 'goAtlanta' ] self.assertEqual( expected_list, keyword_generator.generate_search_terms( '20901970-53a0-417c-b5b4-832a74148af6'))