13 lines
362 B
Python
13 lines
362 B
Python
from pandas import DataFrame
|
|
|
|
from laborious.utils.dataframe_debug import build_dataframe_debug_message
|
|
|
|
|
|
def test_build_dataframe_debug_message_skips_large_dataframe():
|
|
df = DataFrame({'a': [1, 2, 3]})
|
|
|
|
msg = build_dataframe_debug_message('payload', df, max_rows=1)
|
|
|
|
assert 'skipped because dataframe has 3 rows' in msg
|
|
assert '(max: 1)' in msg
|