tgorordo/carousel
1
0
Fork 0
mirror of https://github.com/tgorordo/carousel.git synced 2026-06-12 20:42:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

prelim cli, working core

Browse source
This commit is contained in:
Thomas (Tom) C. Gorordo 2026-05-29 17:50:34 -07:00
parent 39c0ad1f7c
commit f4fb70e23b
Signed by: tgorordo
GPG key ID: 0CBED22BB0D94490
7 changed files with 464 additions and 181 deletions
Download patch file Download diff file Expand all files Collapse all files

80
test/galeshapley_test.py
View file

@ -1,80 +0,0 @@
import polars as pl
from polars.testing import assert_frame_equal
from carousel import GS_deferred_acceptance
def test_prefs():
people_prefs = pl.DataFrame(
{
"people": [
"Alice",
"Alice",
"Alice",
"Bob",
"Bob",
"Bob",
"Charlie",
"Charlie",
"Charlie",
],
"fruit": [
"apple",
"banana",
"cherry",
"banana",
"cherry",
"apple",
"cherry",
"apple",
"banana",
],
"rank": [1, 2, 3, 1, 2, 3, 1, 2, 3],
}
)
fruit_prefs = pl.DataFrame(
{
"fruit": [
"apple",
"apple",
"apple",
"banana",
"banana",
"banana",
"cherry",
"cherry",
"cherry",
],
"people": [
"Alice",
"Bob",
"Charlie",
"Alice",
"Bob",
"Charlie",
"Alice",
"Bob",
"Charlie",
],
"rank": [1, 1, 1, 1, 1, 1, 1, 1, 1], # fruits have no preferences
}
)
capacities = pl.DataFrame(
{
"fruit": ["apple", "cherry", "banana"],
"capacity": [1, 1, 1], # have one of each
}
)
assert_frame_equal(
GS_deferred_acceptance(
people_prefs, fruit_prefs, capacities, app_col="people", pos_col="fruit"
).sort(["people", "fruit"]),
pl.DataFrame(
{
"people": ["Alice", "Bob", "Charlie"],
"fruit": ["apple", "banana", "cherry"],
}
).sort(["people", "fruit"]),
)