def _get_questions_for_survey_page(self, page_num): """ Gets a page of the survey. """ if not page_num: raise ValueError('page_num is required') if type(page_num) != int: page_num = int(page_num) if page_num <= 0: raise ValueError('page_num must be 1 or higher') from_number, to_number = self._calculate_from_and_to_for_page_number(page_num) return Question.get_questions_by_number_range(from_number, to_number)
def test_number_range_21_to_40_returns_20_questions(self): import_questions() quest_range = Question.get_questions_by_number_range(21, 40) self.assertEqual(20, len(quest_range))
def test_to_number_is_required(self): with self.assertRaises(ValueError): Question.get_questions_by_number_range(50, None)