def test_remove_all_columns(two_pod_table): table = two_pod_table remove_columns(table, "*") assert len(table.columns) == 0 assert table.rows[0]["cells"] == []
def test_remove_columns(two_pod_table): table = two_pod_table remove_columns(table, "Status") assert len(table.columns) == 1 assert table.rows[0]["cells"] == ["pod-a"]