def test_view_reprs(): """Test our custom DictView objects.""" table = Table(value=_TABLE_DATA["dict"]) assert repr(table.keys()) == "column_headers(['col_1', 'col_2', 'col_3'])" assert repr(table.keys("column")) == "column_headers(['col_1', 'col_2', 'col_3'])" assert repr(table.keys("row")) == "row_headers(['r1', 'r2'])" assert repr(table.items()) == "table_items(3 columns)" assert repr(table.items("column")) == "table_items(3 columns)" assert repr(table.items("row")) == "table_items(2 rows)"
def test_dataview_delitem(): """Test that table.data can be indexed like a numpy array.""" input = _TABLE_DATA["dict"] table = Table(value=input) row_keys = table.keys("row") # also demoing keys views col_keys = table.keys("column") # also demoing keys views assert list(row_keys) == ["r1", "r2"] assert list(col_keys) == ["col_1", "col_2", "col_3"] del table.data[1] assert not table.to_dict("dict") == input assert list(row_keys) == ["r1"] assert list(col_keys) == ["col_1", "col_2", "col_3"] del table.data[:, 2] assert list(row_keys) == ["r1"] assert list(col_keys) == ["col_1", "col_2"] with pytest.raises(ValueError): del table.data[0, 0] # cannot delete cells