From 29ff1643be1544319716d15e0df03d92ee79e4e9 Mon Sep 17 00:00:00 2001 From: Oliver Hensby Date: Sat, 29 Oct 2022 16:28:10 +0100 Subject: [PATCH] =?UTF-8?q?=E2=9C=85=20Updated=20data=20getter=20tests?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Oliver Hensby 🎨 Ran black formatter Formatted again. Docstring was too long --- tests/test_datagrid.py | 21 +++++++++++++++++++-- 1 file changed, 19 insertions(+), 2 deletions(-) diff --git a/tests/test_datagrid.py b/tests/test_datagrid.py index aa007962..7e329250 100644 --- a/tests/test_datagrid.py +++ b/tests/test_datagrid.py @@ -40,9 +40,26 @@ def test_selections(clear: bool, dataframe: pd.DataFrame) -> None: ] -def test_data_getter(dataframe) -> None: +@pytest.mark.parametrize("drop_rows", [True, False]) +def test_data_getter(drop_rows: bool, dataframe: pd.DataFrame) -> None: + """Testing data getter of DataGrid to check it can be called when there is + data and when there isn't data (i.e. all rows have been deleted in the + dataframe). + + Args: + drop_rows (bool): boolean determining whether to drop rows in dataframe. + dataframe (pd.DataFrame): initial dataframe passed. + """ grid = DataGrid(dataframe) - assert grid.data.equals(dataframe) + if drop_rows: + grid.data = grid.data.drop(["One", "Two", "Three"]) # Drop all rows + grid_data = grid.data # calling data getter + assert list(grid_data.columns) == ["A", "B"] + assert grid_data.index.name == "key" + assert list(grid_data.values) == [] + else: + grid_data = grid.data + assert grid_data.equals(dataframe) def test_data_setter(dataframe) -> None: