def test_supervisor_optimal(
student_names, project_names, supervisor_names, capacities, seed
):
""" Verify that the student allocation algorithm produces a valid,
supervisor-optimal solution to an instance of SA. """
np.random.seed(seed)
students, projects, supervisors = make_players(
student_names, project_names, supervisor_names, capacities
)
matching = student_allocation(
students, projects, supervisors, optimal="supervisor"
)
assert set(projects) == set(matching.keys())
assert all(
[
s in set(students)
for s in {
match for matches in matching.values() for match in matches
}
]
)
for supervisor in supervisors:
old_idx = -np.infty
for student in supervisor.matching:
idx = supervisor.prefs.index(student)
assert idx >= old_idx
old_idx = idx
def test_example_in_docs():
""" Verify the example used in the discussion page of SA. """
student_prefs = {
"A": ["X1", "X2"],
"B": ["Y2", "X2", "Y1"],
"C": ["X1", "Y1", "X2"],
"D": ["Y2", "X1", "Y1"],
"E": ["X1", "Y2", "X2", "Y1"],
}
supervisor_prefs = {
"X": ["B", "C", "A", "E", "D"],
"Y": ["B", "C", "E", "D"],
}
project_supervisors = {"X1": "X", "X2": "X", "Y1": "Y", "Y2": "Y"}
project_capacities = {p: 2 for p in project_supervisors}
supervisor_capacities = {sup: 3 for sup in supervisor_prefs}
game = StudentAllocation.create_from_dictionaries(
student_prefs,
supervisor_prefs,
project_supervisors,
project_capacities,
supervisor_capacities,
)
a, b, c, d, e = game.students
(x1, x2, y1, y2), (x, y) = game.projects, game.supervisors
matching = student_allocation(game.students, game.projects,
game.supervisors)
assert matching == {x1: [c, a], x2: [], y1: [d], y2: [b, e]}
def test_student_optimal(student_names, project_names, supervisor_names,
capacities, seed, clean):
"""Verify that the student allocation algorithm produces a valid,
student-optimal solution to an instance of SA."""
np.random.seed(seed)
students, projects, supervisors = make_players(student_names,
project_names,
supervisor_names,
capacities)
matching = student_allocation(students,
projects,
supervisors,
optimal="student")
assert set(projects) == set(matching.keys())
assert all([
s in set(students)
for s in {match
for matches in matching.values() for match in matches}
])
for student in students:
if student.matching:
assert student.prefs.index(student.matching) == 0