Testing DataTable.get_column with invalid column key

This commit is contained in:
Darren Burns
2023-02-14 13:22:07 +00:00
parent af694ecb97
commit 6d888af723

View File

@@ -423,6 +423,14 @@ async def test_get_column():
next(cells)
async def test_get_column_invalid_key():
app = DataTableApp()
async with app.run_test():
table = app.query_one(DataTable)
with pytest.raises(ColumnDoesNotExist):
list(table.get_column("INVALID"))
async def test_update_cell_cell_exists():
app = DataTableApp()
async with app.run_test():