1 files changed, 7 insertions, 13 deletions
diff --git a/tests/complete_test.py b/tests/complete_test.py
index 769dcd9..c81c424 100755
--- a/tests/complete_test.py
+++ b/tests/complete_test.py
@@ -12,29 +12,23 @@ from quizdata.question import MultipleChoiceQuestion, MissingWordQuestion
 from peak.rules import when
 
 def do_question(q):
-    print "Unhandled question type."
+    print "Unhandled question type.", type(q)
 
 @when(do_question, (MultipleChoiceQuestion,))
 def do_multi_questions(q):
     print plain_text(q.text)
     for a in zip('0123456789', q.answers):
-        print a
-    answer = raw_input()
+        print "%5s: %s" % a
+    answer = int(raw_input())
+    q.answer = q.answers[answer]
     print q.correct
 
 @when(do_question, (MissingWordQuestion,))
-def do_missing_multi_question(q):
-    print dir(q)
-    print plain_text(q.text), '__________',
-    print plain_text(q.tail_text)
-    for a in zip('0123456789', q.answers):
-        print a
-    answer = raw_input()
-    print q.correct
-
+def do_mw_question(q): # XXX: inheritance is annoying here...
+    print "Unhandled question type. (MissingWordQuestion)"
 
 def main():
-    questions = quizdata.open("file://%s?format=gift" % path.join(base_path, 'tests', 'examples.txt'))
+    questions = quizdata.open("file://%s?format=gift" % path.join(base_path, 'tests', 'multi_choice.txt'))
     for q in questions:
         do_question(q)