Python SearchQuerySetWrapper.count Examples

Example #1

File: test_utils.py Project: JioEducation/course-discovery

class SearchQuerySetWrapperTests(TestCase):
    def setUp(self):
        super(SearchQuerySetWrapperTests, self).setUp()
        title = 'Some random course'
        query = 'title:' + title

        CourseRunFactory.create_batch(3, title=title)
        self.search_queryset = SearchQuerySet().models(CourseRun).raw_search(query).load_all()
        self.course_runs = [e.object for e in self.search_queryset]
        self.wrapper = SearchQuerySetWrapper(self.search_queryset)

    def test_count(self):
        self.assertEqual(self.search_queryset.count(), self.wrapper.count())

    def test_iter(self):
        self.assertEqual([e for e in self.course_runs], [e for e in self.wrapper])

    def test_getitem(self):
        self.assertEqual(self.course_runs[0], self.wrapper[0])

Example #2

class SearchQuerySetWrapperTests(TestCase):
    def setUp(self):
        super().setUp()
        title = 'Some random course'
        query = 'title:' + title

        CourseRunFactory.create_batch(3, title=title)
        self.search_queryset = CourseRunDocument().search().query(
            'query_string', query=query)
        self.course_runs = [e.object for e in self.search_queryset]
        self.wrapper = SearchQuerySetWrapper(self.search_queryset)

    def test_count(self):
        self.assertEqual(self.search_queryset.count(), self.wrapper.count())

    def test_iter(self):
        self.assertEqual(self.course_runs, list(self.wrapper))

    def test_getitem(self):
        self.assertEqual(self.course_runs[0], self.wrapper[0])

Example #3

File: test_utils.py Project: regisb/course-discovery

class SearchQuerySetWrapperTests(TestCase):
    def setUp(self):
        super(SearchQuerySetWrapperTests, self).setUp()
        title = 'Some random course'
        query = 'title:' + title

        CourseRunFactory.create_batch(3, title=title)
        self.search_queryset = SearchQuerySet().models(CourseRun).raw_search(
            query).load_all()
        self.course_runs = [e.object for e in self.search_queryset]
        self.wrapper = SearchQuerySetWrapper(self.search_queryset)

    def test_count(self):
        self.assertEqual(self.search_queryset.count(), self.wrapper.count())

    def test_iter(self):
        self.assertEqual(self.course_runs, list(self.wrapper))

    def test_getitem(self):
        self.assertEqual(self.course_runs[0], self.wrapper[0])