From 3b7d24f9c041c0f226ef44945b0ad34ab911a064 Mon Sep 17 00:00:00 2001 From: Darren Burns Date: Tue, 14 Feb 2023 13:25:01 +0000 Subject: [PATCH] Testing DataTable.get_column_at --- tests/test_data_table.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/tests/test_data_table.py b/tests/test_data_table.py index bf072bd0b..2ebbd8380 100644 --- a/tests/test_data_table.py +++ b/tests/test_data_table.py @@ -431,6 +431,17 @@ async def test_get_column_invalid_key(): list(table.get_column("INVALID")) +async def test_get_column_at(): + app = DataTableApp() + async with app.run_test(): + table = app.query_one(DataTable) + table.add_columns("A", "B") + table.add_rows(ROWS) + + first_column = list(table.get_column_at(0)) + assert first_column == [ROWS[0][0], ROWS[1][0], ROWS[2][0]] + + async def test_update_cell_cell_exists(): app = DataTableApp() async with app.run_test():