from pathlib import Path from typing import Callable import pytest # These paths should be relative to THIS directory. SNAPSHOT_APPS_DIR = Path("./snapshot_apps") def test_auto_table(snap_compare: Callable) -> None: assert snap_compare(SNAPSHOT_APPS_DIR / "auto-table.py", terminal_size=(120, 40)) def test_datatable_render(snap_compare: Callable) -> None: press = ["down", "down", "right", "up", "left"] assert snap_compare(SNAPSHOT_APPS_DIR / "data_table.py", press=press) def test_datatable_row_cursor_render(snap_compare: Callable) -> None: press = ["up", "left", "right", "down", "down"] assert snap_compare(SNAPSHOT_APPS_DIR / "data_table_row_cursor.py", press=press) def test_datatable_no_render_markup(snap_compare: Callable) -> None: assert snap_compare(SNAPSHOT_APPS_DIR / "data_table_no_render_markup.py") def test_datatable_null_mixed_cols(snap_compare: Callable) -> None: assert snap_compare(SNAPSHOT_APPS_DIR / "data_table_null_mixed_cols.py") def test_datatable_range_cursor_render(snap_compare: Callable) -> None: press = ["right", "down", "shift+right", "shift+down", "shift+down"] assert snap_compare(SNAPSHOT_APPS_DIR / "data_table_range_cursor.py", press=press) def test_datatable_column_cursor_render(snap_compare: Callable) -> None: press = ["left", "up", "down", "right", "right"] assert snap_compare(SNAPSHOT_APPS_DIR / "data_table_column_cursor.py", press=press) def test_datatable_max_width_render(snap_compare: Callable) -> None: press = ["right", "down", "shift+right", "shift+down", "shift+down"] assert snap_compare(SNAPSHOT_APPS_DIR / "data_table_max_width.py", press=press) def test_datatable_sort_multikey(snap_compare: Callable) -> None: press = ["down", "right", "s"] # Also checks that sort doesn't move cursor. assert snap_compare(SNAPSHOT_APPS_DIR / "data_table_sort.py", press=press) def test_datatable_remove_row(snap_compare: Callable) -> None: press = ["r"] assert snap_compare(SNAPSHOT_APPS_DIR / "data_table_remove_row.py", press=press) @pytest.mark.skip(reason="Don't support row labels.") def test_datatable_labels_and_fixed_data(snap_compare: Callable) -> None: # Ensure that we render correctly when there are fixed rows/cols and labels. assert snap_compare(SNAPSHOT_APPS_DIR / "data_table_row_labels.py") # skip, don't xfail; see: https://github.com/Textualize/pytest-textual-snapshot/issues/6 @pytest.mark.skip( reason=( "The data in this test includes markup; the backend doesn't" "know these have zero width, so we draw the column wider than we used to" ) ) def test_datatable_style_ordering(snap_compare: Callable) -> None: # Regression test for https -> None://github.com/Textualize/textual/issues/2061 assert snap_compare(SNAPSHOT_APPS_DIR / "data_table_style_order.py") def test_datatable_add_column(snap_compare: Callable) -> None: # Checking adding columns after adding rows assert snap_compare(SNAPSHOT_APPS_DIR / "data_table_add_column.py") @pytest.mark.skip(reason="No multi-height rows. No Rich objects.") def test_datatable_add_row_auto_height(snap_compare: Callable) -> None: # Check that rows added with auto height computation look right. assert snap_compare(SNAPSHOT_APPS_DIR / "data_table_add_row_auto_height.py") @pytest.mark.skip(reason="No multi-height rows. No Rich objects.") def test_datatable_add_row_auto_height_sorted(snap_compare: Callable) -> None: # Check that rows added with auto height computation look right. assert snap_compare( SNAPSHOT_APPS_DIR / "data_table_add_row_auto_height.py", press=["s"] ) def test_datatable_empty(snap_compare: Callable) -> None: assert snap_compare(SNAPSHOT_APPS_DIR / "empty.py") def test_datatable_empty_add_col(snap_compare: Callable) -> None: assert snap_compare(SNAPSHOT_APPS_DIR / "empty_add_col.py") def test_datatable_no_rows(snap_compare: Callable) -> None: assert snap_compare(SNAPSHOT_APPS_DIR / "no_rows.py") def test_datatable_no_rows_empty_sequence(snap_compare: Callable) -> None: assert snap_compare(SNAPSHOT_APPS_DIR / "no_rows_empty_sequence.py") def test_datatable_from_parquet(snap_compare: Callable) -> None: assert snap_compare(SNAPSHOT_APPS_DIR / "from_parquet.py") def test_datatable_from_records(snap_compare: Callable) -> None: assert snap_compare(SNAPSHOT_APPS_DIR / "from_records.py") def test_datatable_from_pydict(snap_compare: Callable) -> None: assert snap_compare(SNAPSHOT_APPS_DIR / "from_pydict_with_col_labels.py")