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_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_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_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_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_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_primary_secondary(self):
with self.settings(RESTCLIENTS_SWS_DAO_CLASS=FDAO_SWS):
regid = "00000000000000000000000000000003"
term = Term()
term.year = 2012
term.quarter = "summer"
schedule = _get_schedule(regid, term)
colors = get_colors_by_regid_and_schedule(regid, schedule)
self.assertEquals
(colors["2012,summer,PHYS,121/A"], 1, "Primary gets the 1st color")
msg = "Secondary gets the 1st color, secondary version"
self.assertEquals(colors["2012,summer,PHYS,121/AC"], "1a", msg)
msg = "Second secondary gets the 1st color, secondary version"
self.assertEquals(colors["2012,summer,PHYS,121/AQ"], "1a", msg)
def test_primary_secondary(self):
with self.settings(RESTCLIENTS_SWS_DAO_CLASS=FDAO_SWS):
regid = "00000000000000000000000000000003"
term = Term()
term.year = 2012
term.quarter = "summer"
schedule = _get_schedule(regid, term)
colors = get_colors_by_regid_and_schedule(regid, schedule)
self.assertEquals
(colors["2012,summer,PHYS,121/A"], 1, "Primary gets the 1st color")
msg = "Secondary gets the 1st color, secondary version"
self.assertEquals(
colors["2012,summer,PHYS,121/AC"], "1a", msg)
msg = "Second secondary gets the 1st color, secondary version"
self.assertEquals(
colors["2012,summer,PHYS,121/AQ"], "1a", msg)
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")
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")
