math2b = Course("MATH2B") math2a = Course("MATH2A") ics6n = Course("ICS6N") math3a = Course("MATH3A") ics46 = Course("ICS46") ics45c = Course("ICS45C") ics33 = Course("ICS33") ics32 = Course("ICS32") ics31 = Course("ICS31") cs161 = Course("CS161") stats67 = Course("STATS67") stats7 = Course("STATS7") stats120a = Course("STATS120A") cs178 = Course("CS178") math2b.addPrerequisite(math2a) ics46.addPrerequisite(ics45c) ics45c.addPrerequisite(ics33) ics33.addPrerequisite(ics32) ics32.addPrerequisite(ics31) stats67.addPrerequisite(math2b) cs161.addPrerequisites([ics6b, ics6d, math2b, ics46]) cs178.addPrerequisites([ math2b, [stats67, [stats7, stats120a]], ics6b, ics6d, [math3a, ics6n] ]) course_list = [ ics6b, ics6d, math2b, math2a, stats67, stats7, stats120a, ics6n, math3a, ics46, ics45c, ics33, ics32, ics31, cs161, cs178 ]
v1=y.get_vertices()[0].get_data(), v2=y.get_vertices()[1].get_data()) for y in self._edges ]) if __name__ == "__main__": ics6b = Course("ICS6B") ics6d = Course("ICS6D") math2b = Course("MATH2B") ics6n = Course("ICS6N") math3a = Course("MATH3A") ics46 = Course("ICS46") cs161 = Course("CS161") cs161.addPrerequisite(ics6b) cs161.addPrerequisite(ics6d) cs161.addPrerequisite(math2b) cs161.addPrerequisiteDisjunction([ics6n, math3a]) cs161.addPrerequisite(ics46) g1 = CourseGraph() g1.add_vertex(0, cs161) g1.add_vertex(1, ics6b, [0]) g1.add_vertex(2, ics6d, [0]) g1.add_vertex(3, math2b, [0]) g1.add_vertex(4, ics6n) g1.add_vertex(5, math3a) ics6n_vertex = g1.get_vertex(4) math3a_vertex = g1.get_vertex(5)