Unit testing the decider questions

decider_utils.py

@@ -6,7 +6,7 @@ NO = False
 
 def yesno(question, text, default):
     prompt = text + "\n\n" + question
-
+    print(prompt)
     hopefully_word_yes_or_no = openai.Completion.create(
         engine="text-davinci-002",
         prompt=prompt,
@@ -16,7 +16,7 @@ def yesno(question, text, default):
         presence_penalty=0,
         n=1,
     )["choices"][0]["text"]
-
+    print(hopefully_word_yes_or_no)
     hopefully_word_yes_or_no = hopefully_word_yes_or_no.upper().strip()
 
     result = default

run_unit_tests.py

@@ -0,0 +1,25 @@
+#! /usr/bin/env python3
+
+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_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 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)
+