tests: updating tests for structured grid datastructure

tests/unit/datatypes/test__structured_grid.py

@@ -8,18 +8,23 @@ def test_structured_grid_to_dict():
     origin = np.array([0, 0, 0])
     nsteps = np.array([10, 10, 10])
     step_vector = np.array([1, 1, 1])
-    data = rng.random(size=(10, 10, 10))
+    data = {'rng': rng.random(size=(10, 10, 10))}
     name = "grid_data"
 
     grid = StructuredGrid(
-        origin=origin, step_vector=step_vector, nsteps=nsteps, data=data, name=name
+        origin=origin,
+        step_vector=step_vector,
+        nsteps=nsteps,
+        properties_vertex=data,
+        properties_cell={},
+        name=name,
     )
     grid_dict = grid.to_dict()
 
     assert np.all(grid_dict["origin"] == origin)
     assert np.all(grid_dict["nsteps"] == nsteps)
     assert np.all(grid_dict["step_vector"] == step_vector)
-    assert np.all(grid_dict["data"] == data)
+    assert np.all(grid_dict["properties_vertex"] == data)
     assert grid_dict["name"] == name
 
 
@@ -28,7 +33,14 @@ def test_structured_grid_maximum():
     nsteps = np.array([10, 10, 10])
     step_vector = np.array([1, 1, 1])
 
-    grid = StructuredGrid(origin, step_vector, nsteps, None, None)
+    grid = StructuredGrid(
+        origin=origin,
+        step_vector=step_vector,
+        nsteps=nsteps,
+        properties_cell={},
+        properties_vertex={},
+        name=None,
+    )
     maximum = grid.maximum
 
     expected_maximum = origin + nsteps * step_vector
@@ -43,10 +55,17 @@ def test_structured_grid_vtk():
     origin = np.array([0, 0, 0])
     nsteps = np.array([10, 10, 10])
     step_vector = np.array([1, 1, 1])
-    data = rng.random(size=(10, 10, 10))
+    data = {'rng': rng.random(size=(10, 10, 10))}
     name = "grid_data"
 
-    grid = StructuredGrid(origin, step_vector, nsteps, data, name)
+    grid = StructuredGrid(
+        origin=origin,
+        step_vector=step_vector,
+        nsteps=nsteps,
+        properties_vertex=data,
+        properties_cell={},
+        name=name,
+    )
     vtk_grid = grid.vtk()
     grid_points = vtk_grid.points
     grid_origin = np.min(grid_points, axis=0)
@@ -55,7 +74,7 @@ def test_structured_grid_vtk():
     assert isinstance(vtk_grid, pv.RectilinearGrid)
     assert np.array_equal(vtk_grid.dimensions, nsteps)
     assert np.array_equal(grid_origin, origin)
-    assert np.array_equal(vtk_grid[name], data.flatten(order="F"))
+    assert np.array_equal(vtk_grid['rng'], data['rng'].flatten(order="F"))
 
 
 if __name__ == "__main__":