Unit testing the decider questions

app.py

@@ -83,7 +83,8 @@ def run_1_game_turn(s_narr_transcript, s_n_turns_elapsed, s_user_transcript, s_u
     elif decider_utils.yesno(QUESTION_IS_ACTION_LIKELY_LETHAL, s_user_input, default=NO):
         finally_add2_both_tscripts += game_over_fail_txt("You have taken an action that is likely to result in killing someone.")
 
-    elif decider_utils.yesno(QUESTION_IS_USER_ENGAGED_WITH_BANDITS, s_narr_transcript, default=NO) and decider_utils.yesno(QUESTION_IS_ACTION_RUNNING_AWAY, s_user_input, default=NO):
+    elif decider_utils.yesno(QUESTION_IS_USER_ENGAGED_WITH_BANDITS, s_narr_transcript, default=NO) \
+    and decider_utils.special_case_is_running_away(s_user_input):
         finally_add2_both_tscripts += "Invalid entry.  You cannot outrun these bandits.\n"
 
     elif decider_utils.yesno(QUESTION_IS_ACTION_MAGIC, s_user_input, default=NO):

decider_questions.py

@@ -2,6 +2,6 @@ QUESTION_IS_USER_HOME = "At the end of the above story, is the protagonist locat
 QUESTION_DOES_USER_STILL_HAVE_AT_LEAST_30_GOLD = "At the end of the above story, does the protagonist still have at least 30 gold pieces?"
 QUESTION_IS_USER_ENGAGED_WITH_BANDITS = "At the end of the above story, is the protagonist currently still engaged in a standoff with bandits?"
 QUESTION_IS_ACTION_LIKELY_LETHAL = "Is the action just described likely to result in anyone dying?"
-QUESTION_IS_ACTION_RUNNING_AWAY = 'Does the preceding statement include the words "I run away" or something very similar?'
+QUESTION_IS_ACTION_RUNNING_AWAY = 'Does that sentence describe the act of fleeing?'
 QUESTION_IS_ACTION_MAGIC = "Is the action just described an example of using supernatural magical spells / potions / etc?"
 QUESTION_DID_PROTAGONIST_KILL = "In the story segment above, did the protagonist kill anyone?"

decider_utils.py

@@ -1,4 +1,5 @@
 import openai
+import decider_questions
 
 YES = True
 NO = False
@@ -30,3 +31,18 @@ def yesno(question, text, default):
             result = YES
 
     return result
+
+
+# In certain cases, I need more special-case logic in order to behave correctly,
+# which we verify using the unit tests in run_unit_tests.py:
+
+def special_case_is_running_away(text):
+    might_really_be_fleeing = False
+    for keyword in ["run", "away", "hide", "escape", "flee", "sprint", "teleport"]:
+        if keyword in text.lower():
+            might_really_be_fleeing = True
+
+    if might_really_be_fleeing:
+        return yesno(decider_questions.QUESTION_IS_ACTION_RUNNING_AWAY, text, default=NO)
+    else:
+        return NO

run_unit_tests.py

@@ -1,25 +1,26 @@
 #! /usr/bin/env python3
 
+import os
 import openai
 
 openai.organization = os.environ.get("OPENAI_ORGANIZATION")
 openai.api_key = os.environ.get("OPENAI_KEY")
 
 
-from decider_utils import yesno, YES, NO
+from decider_utils import YES, NO, yesno, special_case_is_running_away
 from decider_questions import *
 
 
 
-assert YES == yesno(QUESTION_IS_ACTION_RUNNING_AWAY, "run away", default=NO)
-assert YES == yesno(QUESTION_IS_ACTION_RUNNING_AWAY, "I run away", default=NO)
-assert YES == yesno(QUESTION_IS_ACTION_RUNNING_AWAY, "flee", default=NO)
-assert YES == yesno(QUESTION_IS_ACTION_RUNNING_AWAY, "I flee.", default=NO)
+assert YES == special_case_is_running_away("run away")
+assert YES == special_case_is_running_away("I run away")
+assert YES == special_case_is_running_away("flee")
+assert YES == special_case_is_running_away("I flee.")
 
-assert NO == yesno(QUESTION_IS_ACTION_RUNNING_AWAY, "A", default=NO)
-assert NO == yesno(QUESTION_IS_ACTION_RUNNING_AWAY, "B", default=NO)
-assert NO == yesno(QUESTION_IS_ACTION_RUNNING_AWAY, "C", default=NO)
-assert NO == yesno(QUESTION_IS_ACTION_RUNNING_AWAY, "ok", default=NO)
-assert NO == yesno(QUESTION_IS_ACTION_RUNNING_AWAY, "say ok", default=NO)
-assert NO == yesno(QUESTION_IS_ACTION_RUNNING_AWAY, "say okay", default=NO)
+assert NO == special_case_is_running_away("A")
+assert NO == special_case_is_running_away("B")
+assert NO == special_case_is_running_away("C")
+assert NO == special_case_is_running_away("ok")
+assert NO == special_case_is_running_away("say ok")
+assert NO == special_case_is_running_away("say okay")