From b5c9f8e862caef2b7e4e0122827a608a33c5071a Mon Sep 17 00:00:00 2001
From: Peter Pinch <pdpinch@mit.edu>
Date: Fri, 18 May 2018 22:31:47 -0400
Subject: [PATCH] fix tests

---
 .../lib/xmodule/xmodule/tests/test_capa_module.py  | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/common/lib/xmodule/xmodule/tests/test_capa_module.py b/common/lib/xmodule/xmodule/tests/test_capa_module.py
index 21c61cedf1..08cf8f37d8 100644
--- a/common/lib/xmodule/xmodule/tests/test_capa_module.py
+++ b/common/lib/xmodule/xmodule/tests/test_capa_module.py
@@ -275,39 +275,39 @@ class CapaModuleTest(unittest.TestCase):
             'showanswer': 'attempted',
             'max_attempts': '1',
             'show_correctness': 'always',
-        }, True),
+        }, False, True),
         # If show_correctness=never, Answer is never visible
         ({
             'showanswer': 'attempted',
             'max_attempts': '1',
             'show_correctness': 'never',
-        }, False),
+        }, False, False),
         # If show_correctness=past_due, answer is not visible before due date
         ({
             'showanswer': 'attempted',
             'show_correctness': 'past_due',
             'max_attempts': '1',
             'due': 'tomorrow_str',
-        }, False),
+        }, False, False),
         # If show_correctness=past_due, answer is visible after due date
         ({
             'showanswer': 'attempted',
             'show_correctness': 'past_due',
             'max_attempts': '1',
             'due': 'yesterday_str',
-        }, True),
+        }, True, True),
     )
     @ddt.unpack
-    def test_showanswer_hide_correctness(self, problem_data, answer_available):
+    def test_showanswer_hide_correctness(self, problem_data, answer_available_no_attempt, answer_available_after_attempt):
         """
         Ensure that the answer will not be shown when correctness is being hidden.
         """
         if 'due' in problem_data:
             problem_data['due'] = getattr(self, problem_data['due'])
         problem = CapaFactory.create(**problem_data)
-        self.assertFalse(problem.answer_available())
+        self.assertEqual(problem.answer_available(), answer_available_no_attempt)
         problem.attempts = 1
-        self.assertEqual(problem.answer_available(), answer_available)
+        self.assertEqual(problem.answer_available(), answer_available_after_attempt)
 
     def test_showanswer_closed(self):