test_show.py

from inspect import signature
from unittest.mock import patch

import pytest

import polars as pl


def test_show_signature_match() -> None:
    assert signature(pl.LazyFrame.show) == signature(pl.DataFrame.show)


def test_lf_show_calls_df_show() -> None:
    lf = pl.LazyFrame({})
    with patch.object(pl.DataFrame, "show") as df_show:
        lf.show(5)

    df_show.assert_called_once_with(
        5,
        ascii_tables=None,
        decimal_separator=None,
        thousands_separator=None,
        float_precision=None,
        fmt_float=None,
        fmt_str_lengths=None,
        fmt_table_cell_list_len=None,
        tbl_cell_alignment=None,
        tbl_cell_numeric_alignment=None,
        tbl_cols=None,
        tbl_column_data_type_inline=None,
        tbl_dataframe_shape_below=None,
        tbl_formatting=None,
        tbl_hide_column_data_types=None,
        tbl_hide_column_names=None,
        tbl_hide_dtype_separator=None,
        tbl_hide_dataframe_shape=None,
        tbl_width_chars=None,
        trim_decimal_zeros=True,
    )


def test_lf_show_no_limit_raises() -> None:
    lf = pl.LazyFrame({})
    with pytest.raises(ValueError):
        lf.show(limit=None)