class Test(unittest.TestCase):

    def setUp(self):
        self.ctq = CourseTableQuery({}, 20)
        self.ctq.connect()
        self.ctq.set_cursor(self.ctq.course_query.get_courses_cursor_subject(subject_code))
        
    def tearDown(self):
        self.ctq.disconnect()

    def test1(self):
        jsonlist = self.ctq.get_table_page_JSON_list(1)
        self.assertEqual(len(jsonlist), 20)
        for json in jsonlist:
            self.assertEqual(json['code'], 'CS')
        
    def test2(self):
        cursor1 = self.ctq.get_table_page_cursor(1)
        cq = CourseQuery()
        cq.connect()
        cursor2 = cq.get_courses_cursor_subject(subject_code)
        cq.disconnect()
        list1 = []
        for x in cursor1:
            list1.append(x)
        self.assertEqual(len(list1), 20)
        list2 = []
        for x in cursor2:
            list2.append(x)
        for a,b in zip(list1,list2):
            self.assertEqual(a,b)
 def setUp(self):
     self.ctq = CourseTableQuery({}, 20)
     self.ctq.connect()
     self.ctq.set_cursor(self.ctq.course_query.get_courses_cursor_subject(subject_code))
Esempio n. 3
0
 client_courses.disconnect()
 
 
 section_query = SectionQuery()
 section_query.connect()
 print(section_query.get_cursor({})[0])
 print(section_query.client[section_query.db_name][section_query.collection_name].distinct('days_of_week'))
 print('HERE')
 section_query.disconnect()
 
 prof_query = ProfQuery()
 prof_query.connect()
 print(prof_query.get_cursor({})[0])
 prof_query.disconnect()
 
 table_query = CourseTableQuery({}, 20)
 table_query.connect()
 print(table_query.get_table_page_JSON_list(1))
 table_query.disconnect()
 
 course_query = CourseQuery()
 course_query.connect()
 print(course_query.client[course_query.db_name][course_query.collection_name])
 print(sorted(course_query.client[course_query.db_name][course_query.collection_name].distinct('code')))
 print(course_query.get_course_JSON('CS', '125'))
 print(course_query.get_subject_codes())
 print(course_query.get_course_ids_for_subject('CS'))
 #print(course_query.get_course_ids_all_subjects())
 course_query.disconnect()
 
 course_query.connect()