def test_098(self): term = Term() term.year = 2012 term.quarter = "autumn" person = Person() person.uwnetid = "javerage" person.regid = "A0000000000000000000000000000001" schedule = ClassSchedule() schedule.term = term schedule.user = person schedule.sections = [] section = Section() section.term = term section.curriculum_abbr = "MATH" section.course_number = "098" section.section_id = "A" section.is_primary_section = True schedule.sections.append(section) colors = get_colors_by_regid_and_schedule(person.regid, schedule) self.assertEquals (colors[section.section_label()], 1, "one section gets the 1st color")
def test_drop_then_new_add(self): term = Term() term.year = 2012 term.quarter = "autumn" person = Person() person.uwnetid = "javerage" person.regid = "00000000000000000000000000000004" schedule = ClassSchedule() schedule.term = term schedule.user = person schedule.sections = [] section = Section() section.term = term section.curriculum_abbr = "MATH" section.course_number = 124 section.section_id = "A" section.is_primary_section = True schedule.sections.append(section) section2 = Section() section2.term = term section2.curriculum_abbr = "MATH" section2.course_number = 300 section2.section_id = "A" section2.is_primary_section = True schedule.sections.append(section2) # Get the colors for the first round - # doesn't matter what they are though colors = get_colors_by_regid_and_schedule(person.regid, schedule) schedule.sections = [] schedule.sections.append(section) section3 = Section() section3.term = term section3.curriculum_abbr = "MATH" section3.course_number = 309 section3.section_id = "A" section3.is_primary_section = True schedule.sections.append(section3) colors = get_colors_by_regid_and_schedule(person.regid, schedule) self.assertEquals (colors[section.section_label()], 1, "1st section gets the 1st color") self.assertNotIn(section2.section_label, colors, "There shouldn't be a color for section 2") self.assertEquals (colors[section3.section_label()], 3, "3rd section gets the 3rd color")
def test_swap_order(self): term = Term() term.year = 2012 term.quarter = "autumn" person = Person() person.uwnetid = "javerage" person.regid = "00000000000000000000000000000005" schedule = ClassSchedule() schedule.term = term schedule.user = person schedule.sections = [] section = Section() section.term = term section.curriculum_abbr = "MATH" section.course_number = 124 section.section_id = "A" section.is_primary_section = True schedule.sections.append(section) section2 = Section() section2.term = term section2.curriculum_abbr = "MATH" section2.course_number = 300 section2.section_id = "A" section2.is_primary_section = True schedule.sections.append(section2) colors = get_colors_by_regid_and_schedule(person.regid, schedule) schedule.sections = [] schedule.sections.append(section2) schedule.sections.append(section) colors = get_colors_by_regid_and_schedule(person.regid, schedule) self.assertEquals (colors[section.section_label()], 1, "1st section gets the 1st color") self.assertEquals (colors[section2.section_label()], 2, "2nd section gets the 2nd color")
def test_drop_then_new_add(self): term = Term() term.year = 2012 term.quarter = "autumn" person = Person() person.uwnetid = "javerage" person.regid = "00000000000000000000000000000004" schedule = ClassSchedule() schedule.term = term schedule.user = person schedule.sections = [] section = Section() section.term = term section.curriculum_abbr = "MATH" section.course_number = 124 section.section_id = "A" section.is_primary_section = True schedule.sections.append(section) section2 = Section() section2.term = term section2.curriculum_abbr = "MATH" section2.course_number = 300 section2.section_id = "A" section2.is_primary_section = True schedule.sections.append(section2) # Get the colors for the first round - # doesn't matter what they are though colors = get_colors_by_regid_and_schedule(person.regid, schedule) schedule.sections = [] schedule.sections.append(section) section3 = Section() section3.term = term section3.curriculum_abbr = "MATH" section3.course_number = 309 section3.section_id = "A" section3.is_primary_section = True schedule.sections.append(section3) colors = get_colors_by_regid_and_schedule(person.regid, schedule) self.assertEquals (colors[section.section_label()], 1, "1st section gets the 1st color") self.assertNotIn(section2.section_label(), colors, "There shouldn't be a color for section 2") self.assertEquals (colors[section3.section_label()], 3, "3rd section gets the 3rd color")
def test_over_8(self): term = Term() term.year = 2012 term.quarter = "autumn" person = Person() person.uwnetid = "javerage" person.regid = "00000000000000000000000000000007" schedule = ClassSchedule() schedule.term = term schedule.user = person schedule.sections = [] section = Section() section.term = term section.curriculum_abbr = "MATH" section.course_number = 124 section.section_id = "A" section.is_primary_section = True schedule.sections.append(section) section2 = Section() section2.term = term section2.curriculum_abbr = "MATH" section2.course_number = 300 section2.section_id = "A" section2.is_primary_section = True schedule.sections.append(section2) section3 = Section() section3.term = term section3.curriculum_abbr = "MATH" section3.course_number = 301 section3.section_id = "A" section3.is_primary_section = True schedule.sections.append(section3) section4 = Section() section4.term = term section4.curriculum_abbr = "MATH" section4.course_number = 302 section4.section_id = "A" section4.is_primary_section = True schedule.sections.append(section4) section5 = Section() section5.term = term section5.curriculum_abbr = "MATH" section5.course_number = 303 section5.section_id = "A" section5.is_primary_section = True schedule.sections.append(section5) section6 = Section() section6.term = term section6.curriculum_abbr = "MATH" section6.course_number = 304 section6.section_id = "A" section6.is_primary_section = True schedule.sections.append(section6) section7 = Section() section7.term = term section7.curriculum_abbr = "MATH" section7.course_number = 307 section7.section_id = "A" section7.is_primary_section = True schedule.sections.append(section7) section8 = Section() section8.term = term section8.curriculum_abbr = "MATH" section8.course_number = 308 section8.section_id = "A" section8.is_primary_section = True schedule.sections.append(section8) section9 = Section() section9.term = term section9.curriculum_abbr = "MATH" section9.course_number = 309 section9.section_id = "A" section9.is_primary_section = True schedule.sections.append(section9) section10 = Section() section10.term = term section10.curriculum_abbr = "MATH" section10.course_number = 310 section10.section_id = "A" section10.is_primary_section = True schedule.sections.append(section10) colors = get_colors_by_regid_and_schedule(person.regid, schedule) self.assertEquals(len(colors), 10, "has 10 colors") self.assertEquals (colors[section.section_label()], 1, "1st section gets the 1st color") self.assertEquals (colors[section2.section_label()], 2, "2nd section gets the 2nd color") self.assertEquals (colors[section3.section_label()], 3, "3rd section gets the 3rd color") self.assertEquals (colors[section4.section_label()], 4, "4th section gets the 4th color") self.assertEquals (colors[section4.section_label()], 4, "4th section gets the 4th color") self.assertEquals (colors[section5.section_label()], 5, "5th section gets the 5th color") self.assertEquals (colors[section6.section_label()], 6, "6th section gets the 6th color") self.assertEquals (colors[section7.section_label()], 7, "7th section gets the 7th color") self.assertEquals (colors[section8.section_label()], 8, "8th section gets the 8th color") self.assertEquals (colors[section9.section_label()], 1, "9th section gets the 1st color") self.assertEquals (colors[section10.section_label()], 2, "10th section gets the 2nd color")
def test_around_the_top(self): """ go over the 8 color limit, and reuse blanks """ term = Term() term.year = 2012 term.quarter = "autumn" person = Person() person.uwnetid = "javerage" person.regid = "00000000000000000000000000000006" schedule = ClassSchedule() schedule.term = term schedule.user = person schedule.sections = [] section = Section() section.term = term section.curriculum_abbr = "MATH" section.course_number = 124 section.section_id = "A" section.is_primary_section = True schedule.sections.append(section) section2 = Section() section2.term = term section2.curriculum_abbr = "MATH" section2.course_number = 300 section2.section_id = "A" section2.is_primary_section = True schedule.sections.append(section2) section3 = Section() section3.term = term section3.curriculum_abbr = "MATH" section3.course_number = 301 section3.section_id = "A" section3.is_primary_section = True schedule.sections.append(section3) section4 = Section() section4.term = term section4.curriculum_abbr = "MATH" section4.course_number = 302 section4.section_id = "A" section4.is_primary_section = True schedule.sections.append(section4) section5 = Section() section5.term = term section5.curriculum_abbr = "MATH" section5.course_number = 303 section5.section_id = "A" section5.is_primary_section = True schedule.sections.append(section5) section6 = Section() section6.term = term section6.curriculum_abbr = "MATH" section6.course_number = 304 section6.section_id = "A" section6.is_primary_section = True schedule.sections.append(section6) section7 = Section() section7.term = term section7.curriculum_abbr = "MATH" section7.course_number = 307 section7.section_id = "A" section7.is_primary_section = True schedule.sections.append(section7) colors = get_colors_by_regid_and_schedule(person.regid, schedule) schedule.sections = [] schedule.sections.append(section) schedule.sections.append(section7) colors = get_colors_by_regid_and_schedule(person.regid, schedule) self.assertEquals(len(colors), 2, "Only has colors for 2 courses") self.assertEquals (colors[section.section_label()], 1, "1st section gets the 1st color") self.assertEquals (colors[section7.section_label()], 7, "7th section gets the 7th color") section8 = Section() section8.term = term section8.curriculum_abbr = "MATH" section8.course_number = 308 section8.section_id = "A" section8.is_primary_section = True schedule.sections.append(section8) section9 = Section() section9.term = term section9.curriculum_abbr = "MATH" section9.course_number = 309 section9.section_id = "A" section9.is_primary_section = True schedule.sections.append(section9) colors = get_colors_by_regid_and_schedule(person.regid, schedule) self.assertEquals(len(colors), 4, "Only has colors for 4 courses") self.assertEquals (colors[section.section_label()], 1, "1st section gets the 1st color") self.assertEquals (colors[section7.section_label()], 7, "7th section gets the 7th color") self.assertEquals (colors[section8.section_label()], 8, "8th section gets the 8th color") self.assertEquals (colors[section9.section_label()], 2, "9th section gets the 2nd color") schedule.sections.append(section6) colors = get_colors_by_regid_and_schedule(person.regid, schedule) self.assertEquals(len(colors), 5, "Only has colors for 5 courses") self.assertEquals (colors[section.section_label()], 1, "1st section gets the 1st color") self.assertEquals (colors[section7.section_label()], 7, "7th section gets the 7th color") self.assertEquals (colors[section8.section_label()], 8, "8th section gets the 8th color") self.assertEquals (colors[section9.section_label()], 2, "0th section gets the 2nd color") self.assertEquals (colors[section6.section_label()], 3, "6th section gets the 3rd color when readded")