alihan/textual
1
0
Fork 0
mirror of https://github.com/Textualize/textual.git synced 2025-10-17 02:38:12 +03:00
Code Issues Packages Projects Releases Wiki Activity

Testing DataTable.get_column

Browse Source
This commit is contained in:
Darren Burns
2023-02-14 13:18:19 +00:00
parent 16c9f15ab5
commit af694ecb97
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/test_data_table.py
View File

@@ -376,7 +376,7 @@ async def test_get_row_invalid_row_key():
async with app.run_test():
table = app.query_one(DataTable)
with pytest.raises(RowDoesNotExist):
table.get_row("abc")
table.get_row("INVALID")
async def test_get_row_at():
@@ -409,6 +409,20 @@ async def test_get_row_at_invalid_index(index):
table.get_row_at(index)
async def test_get_column():
app = DataTableApp()
async with app.run_test():
table = app.query_one(DataTable)
a, b = table.add_columns("A", "B")
table.add_rows(ROWS)
cells = table.get_column(a)
assert next(cells) == ROWS[0][0]
assert next(cells) == ROWS[1][0]
assert next(cells) == ROWS[2][0]
with pytest.raises(StopIteration):
next(cells)
async def test_update_cell_cell_exists():
app = DataTableApp()
async with app.run_test():