From a2186d263aa8927a47fdb9f3a23ad39841b64b33 Mon Sep 17 00:00:00 2001
From: adityacp
Date: Thu, 12 Mar 2020 11:30:51 +0530
Subject: Fix test to get lesson or quiz for a unit

---
 yaksh/test_models.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'yaksh/test_models.py')

diff --git a/yaksh/test_models.py b/yaksh/test_models.py
index 6e5a26d..a60a1d6 100644
--- a/yaksh/test_models.py
+++ b/yaksh/test_models.py
@@ -322,8 +322,12 @@ class LearningUnitTestCases(unittest.TestCase):
     def test_learning_unit(self):
         self.assertEqual(self.learning_unit_one.type, 'lesson')
         self.assertEqual(self.learning_unit_two.type, 'quiz')
-        self.assertEqual(self.learning_unit_one.lesson, self.lesson)
-        self.assertEqual(self.learning_unit_two.quiz, self.quiz)
+        self.assertEqual(
+            self.learning_unit_one.get_lesson_or_quiz(), self.lesson
+        )
+        self.assertEqual(
+            self.learning_unit_two.get_lesson_or_quiz(), self.quiz
+        )
         self.assertIsNone(self.learning_unit_one.quiz)
         self.assertIsNone(self.learning_unit_two.lesson)
         self.assertTrue(self.learning_unit_one.check_prerequisite)
-- 
cgit