smithy/test/test_nb.py

import marimo

__generated_with = "0.23.6"
app = marimo.App(width="medium")


@app.cell
def _():
    import marimo as mo
    from pathlib import Path

    return (mo,)


@app.cell
def _():
    import polars as pl
    from smithy import smith_set

    return pl, smith_set


@app.cell
def _(mo, pl):
    df = pl.read_csv(mo.notebook_dir() / "test_ballot.csv")
    df = df.with_columns(
        [
            pl.col(c)  # make safe, clean up
            .cast(pl.Utf8)
            .str.strip_chars()
            .cast(pl.Int64, strict=False)
            .fill_null(df.width + 1)
            for c in df.columns
        ]
    )
    df
    return (df,)


@app.cell
def _(df, smith_set):
    smith_set(df)  # find the smith set (should be "Alice" and "Bob" as a pair)
    return


@app.cell
def _():
    return


if __name__ == "__main__":
    app.run()