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

init commit. modeled on tgorordo/smithy, still lots TODO

Browse source
This commit is contained in:
Thomas (Tom) C. Gorordo 2026-05-29 07:10:47 -07:00
commit 8bc048c0ee
Signed by: tgorordo
GPG key ID: 0CBED22BB0D94490
14 changed files with 2659 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

80
test/galeshapley_test.py Normal file
View file

@ -0,0 +1,80 @@
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"]),
)