add more backend python tests

Signed-off-by: Itay Dafna <[email protected]>

Author: ibdafna

tests/test_datagrid.py

@@ -4,11 +4,21 @@
 from ipydatagrid import DataGrid
 
 
+@pytest.fixture
+def dataframe() -> None:
+    return pd.DataFrame(
+        data={"A": [1, 2, 3], "B": [4, 5, 6]}, index=["One", "Two", "Three"]
+    )
+
+
+@pytest.fixture
+def datagrid(dataframe) -> None:
+    return DataGrid(dataframe)
+
+
 @pytest.mark.parametrize("clear", [True, False])
-def test_selections(clear: bool) -> None:
-    df = pd.DataFrame(data={"A": [1, 2, 3], "B": [4, 5, 6]})
-    layout = {"height": "100px"}
-    grid = DataGrid(df, selection_mode="cell", layout=layout, editable=True)
+def test_selections(clear: bool, dataframe: pd.DataFrame) -> None:
+    grid = DataGrid(dataframe, selection_mode="cell", editable=True)
     grid.select(1, 0, 2, 1)  # Select 1A to 2B
     if clear:
         grid.clear_selection()
@@ -20,3 +30,75 @@ def test_selections(clear: bool) -> None:
             {"c": 0, "r": 2},
             {"c": 1, "r": 2},
         ]
+
+
+def test_data_getter(dataframe) -> None:
+    grid = DataGrid(dataframe)
+    assert grid.data.equals(dataframe)
+
+
+def test_data_setter(dataframe) -> None:
+    grid = DataGrid(dataframe)
+    new_df = pd.DataFrame(data={"A": [0, 0, 0], "B": [4, 5, 6]})
+    grid.data = new_df
+    assert grid.data.equals(new_df)
+
+
+def test_get_cell_value(datagrid: DataGrid) -> None:
+    cell = datagrid.get_cell_value("B", "Three")
+    assert cell == [6]
+
+
+def test_set_cell_value() -> None:
+    """ Cannot be tested without a running front end """
+    pass
+
+
+def get_cell_value_by_index(datagrid: DataGrid) -> None:
+    cell = datagrid.get_cell_value_by_index("B", 1)
+    assert cell == [5]
+
+
+def test_set_cell_value_by_index() -> None:
+    """ Cannot be tested without a running front end """
+    pass
+
+
+@pytest.mark.parametrize("invalid_index", [True, False])
+def test_column_name_to_index(invalid_index: bool, datagrid: DataGrid) -> None:
+    if invalid_index:
+        assert datagrid._column_name_to_index("Z") is None
+    else:
+        assert datagrid._column_name_to_index("A") == 0
+
+
+@pytest.mark.parametrize("invalid_index", [True, False])
+def test_column_index_to_name(invalid_index: bool, datagrid: DataGrid) -> None:
+    if invalid_index:
+        assert datagrid._column_index_to_name(4) is None
+    else:
+        assert datagrid._column_index_to_name(1) == "B"
+
+
+def test_get_col_headers(datagrid) -> None:
+    assert datagrid._get_col_headers() == ["A", "B"]
+
+
+@pytest.mark.parametrize("invalid_prim_key", [True, False])
+def test_get_row_index_of_primary_key(
+    invalid_prim_key: bool, datagrid: DataGrid
+) -> None:
+    if invalid_prim_key:
+        assert datagrid._get_row_index_of_primary_key("Nay") == []
+    else:
+        assert datagrid._get_row_index_of_primary_key("Two") == [1]
+
+
+@pytest.mark.parametrize("invalid_coords", [True, False])
+def test_get_cell_value_by_numerical_index(
+    invalid_coords: bool, datagrid: DataGrid
+) -> None:
+    if invalid_coords:
+        assert datagrid._get_cell_value_by_numerical_index(2, 2) is None
+    else:
+        assert datagrid._get_cell_value_by_numerical_index(1, 0) == 4