refactor: accept code_reginement

Co-authored-by: sourcery-ai[bot] <58596630+sourcery-ai[bot]@users.noreply.github.com>
lisphilar · Apr 26, 2024 · 3b4a9b5 · 3b4a9b5
1 parent fb67e37
commit 3b4a9b5
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/py-polars/polars/dataframe/frame.py b/py-polars/polars/dataframe/frame.py
@@ -1291,9 +1291,11 @@ def _repr_html_(self, *, _from_series: bool = False) -> str:
             max_rows = self.height
 
         overall_alignment = os.environ.get("POLARS_FMT_TABLE_CELL_ALIGNMENT", "RIGHT")
-        if overall_alignment not in {"LEFT", "CENTER", "RIGHT"}:
-            msg = f"invalid table cell alignment: {overall_alignment!r}"
-            raise ValueError(msg)
+def validate_alignment(alignment, valid_options={"LEFT", "CENTER", "RIGHT"}):
+    if alignment not in valid_options:
+        raise ValueError(f"invalid table cell alignment: {alignment!r}")
+
+validate_alignment(overall_alignment)
 
         numeric_alignment = os.environ.get(
             "POLARS_FMT_TABLE_CELL_NUMERIC_ALIGNMENT",