diff --git a/src/carousel/__init__.py b/src/carousel/__init__.py index d2d3b13..56b1f4d 100644 --- a/src/carousel/__init__.py +++ b/src/carousel/__init__.py @@ -1,9 +1,18 @@ -import rich +import logging, rich +from rich.logging import RichHandler + +import itertools as it import polars as pl import polars.selectors as pls -import itertools as it +logging.basicConfig( + level="NOTSET", + format="%(message)s", + datefmt="[%X]", + handlers=[RichHandler(rich_tracebacks=True, tracebacks_suppress=[pl, pls])], +) +log = logging.getLogger("rich") def rank_to_pref(ranking): diff --git a/test/galeshapley_test.py b/test/galeshapley_test.py index 6afb1ff..b485db4 100644 --- a/test/galeshapley_test.py +++ b/test/galeshapley_test.py @@ -1,15 +1,29 @@ -import rich +import logging, rich +from rich.logging import RichHandler + import polars as pl import polars.selectors as pls import numpy as np -rng = np.random.default_rng() - from polars.testing import assert_frame_equal import pytest from hypothesis import given, strategies as st +import carousel as crsl + + +logging.basicConfig( + level="NOTSET", + format="%(message)s", + datefmt="[%X]", + handlers=[RichHandler(rich_tracebacks=True, tracebacks_suppress=[np, pl, pls])], +) + +log = logging.getLogger("rich") + +rng = np.random.default_rng() + @st.composite def rankings(draw, names=["a", "b", "c"], choices=["A", "B", "C"]): @@ -31,8 +45,6 @@ def preferences(draw, names=["a", "b", "c"], choices=["A", "B", "C"]): return p -import carousel as crsl - p = pl.DataFrame({"a": ["A", "C", "B"], "b": ["B", "A", "C"], "c": ["C", "B", "A"]}) r = pl.DataFrame({"": ["A", "B", "C"], "a": [1, 3, 2], "b": [2, 1, 3], "c": [3, 2, 1]})