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

some formatting and add cli compilation

Browse source
This commit is contained in:
Thomas (Tom) C. Gorordo 2026-05-20 12:04:17 -07:00
parent 8836c49091
commit 4a624a4847
Signed by: tgorordo
GPG key ID: 0CBED22BB0D94490
9 changed files with 115 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

56
test/test_rcvs.py
View file

@ -1,35 +1,41 @@
import polars as pl
from smithy import smith_set
def test_condorcet():
df = pl.DataFrame({
'A': [1, 1, 2, 1],
'B': [2, 2, 1, 2],
'C': [3, 3, 3, 3],
})
assert smith_set(df) == ['A']
df = pl.DataFrame(
{
"A": [1, 1, 2, 1],
"B": [2, 2, 1, 2],
"C": [3, 3, 3, 3],
}
)
assert smith_set(df) == ["A"]
def test_rockpprscrcycle():
df = pl.DataFrame({
'A': [1, 2, 3],
'B': [2, 3, 1],
'C': [3, 1, 2],
})
assert smith_set(df) == ['A', 'B', 'C']
df = pl.DataFrame(
{
"A": [1, 2, 3],
"B": [2, 3, 1],
"C": [3, 1, 2],
}
)
assert smith_set(df) == ["A", "B", "C"]
def test_abpair():
df = pl.DataFrame({
"A": [1, 2, 1, 3],
"B": [2, 1, 3, 1],
"C": [3, 3, 2, 2]
})
assert smith_set(df) == ['A', 'B']
df = pl.DataFrame({"A": [1, 2, 1, 3], "B": [2, 1, 3, 1], "C": [3, 3, 2, 2]})
assert smith_set(df) == ["A", "B"]
def test_fourcycle():
df = pl.DataFrame({
"A": [1,2,3,4],
"B": [2,3,4,1],
"C": [3,4,1,2],
"D": [4,1,2,3],
})
assert smith_set(df) == ['A', 'B', 'C', 'D']
df = pl.DataFrame(
{
"A": [1, 2, 3, 4],
"B": [2, 3, 4, 1],
"C": [3, 4, 1, 2],
"D": [4, 1, 2, 3],
}
)
assert smith_set(df) == ["A", "B", "C", "D"]