def test_switch_language_ru(self):
message = self._create_message()
save_chat_language(message, "ru")
self.add_dialog_flow(message)
message["text"] = 'смени язык на английский'
self.assertEqual(get_response(message)[0], 'Меняю язык на английский')
self.assertEqual(get_chat_language(message), "en")
def test_switch_language_en(self):
message = self._create_message()
save_chat_language(message, "en")
self.add_dialog_flow(message)
message["text"] = 'change to russian'
self.assertEqual(
get_response(message)[0], 'I will switch to russian now')
self.assertEqual(get_chat_language(message), "ru")
def test_mood_en_language_amazing_negation(self):
message = self._create_message()
save_chat_language(message, "en")
message["text"] = "I don't feel amazing"
self.add_dialog_flow(message)
self.assertIn(
get_response(message)[0],
["What a pity", "I'm sorry to hear that.", "So sad."])
def test_mood_en_language_amazing(self):
message = self._create_message()
save_chat_language(message, "en")
message["text"] = "I feel amazing now because this shit works"
self.add_dialog_flow(message)
self.assertIn(
get_response(message)[0],
["Nice to hear that!", "I'm happy for you!", "Great news!"])
def test_initial_conversation_en(self):
message = self._create_message()
save_chat_language(message, "en")
message["last_move"] = "some_move_long_time_ago"
self.add_dialog_flow(message)
message["text"] = 'hi'
self.assertIn(
get_response(message)[0], [
"Hello!", "Hi there!", "What's up?", "Nice to see you!",
"Hello dear!"
])
def act_on_move(message, move):
if "language" in move and move["language"]:
lang = move["language"]
save_chat_language(message=message, language=lang)
return None
if "mood" in move and move["mood"]:
save_mood(message, move["mood"])
return None
if "get_mood" in move and move["get_mood"]:
moods = get_moods_for_week(message)
return _compute_percentage(moods)
def test_get_mood_ru(self):
message = self._create_message()
save_chat_language(message, "ru")
message["text"] = "я устал"
self.add_dialog_flow(message)
get_response(message)
message = self._create_message()
save_chat_language(message, "ru")
message["text"] = "как я себя последнее время чувствовал?"
self.add_dialog_flow(message)
self.assertTrue("100% плохое" in get_response(message)[0])
def test_initial_conversation_ru(self):
message = self._create_message()
save_chat_language(message, "ru")
self.add_dialog_flow(message)
message["text"] = 'привет'
response, move = get_response(message)
self.assertIn(response, [
"Приветик :) Давай начнем. Как тебя зовут?",
"Привет! Для начала скажи мне, как тебя зовут",
"Привет, добро пожаловать! Как мне тебя называть?"
])
message = self._create_message()
save_chat_language(message, "ru")
self.add_dialog_flow(message)
message["last_move"] = move["move"]
message["text"] = 'меня зовут Артем а тебя'
self.assertIn(
get_response(message)[0], [
"А я Псих. Приятно познакомиться. Я немного разбираюсь в человеческих чувствах, а еще я запомню все, "
"что ты мне скажешь. Как думаешь, как я тебе буду полезен?"
])
def test_help_en(self):
message = self._create_message()
save_chat_language(message, "en")
self.add_dialog_flow(message)
message["text"] = 'help'
self.assertTrue("Sure," in get_response(message)[0])