From 3275b0307f58144f0bceb7aa41b4c1535c260e69 Mon Sep 17 00:00:00 2001 From: Mark Piper Date: Wed, 2 Oct 2024 14:59:04 -0600 Subject: [PATCH] Clean lint --- tests/framework/test_bmi_ugrid.py | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/tests/framework/test_bmi_ugrid.py b/tests/framework/test_bmi_ugrid.py index 7044266..4d5f89d 100644 --- a/tests/framework/test_bmi_ugrid.py +++ b/tests/framework/test_bmi_ugrid.py @@ -79,8 +79,8 @@ def test_points_grid(): assert isinstance(grid, xr.Dataset) assert grid.ndim == 2 assert grid.metadata["type"] == bmi.grid_type(grid_id) - assert grid.data_vars["mesh"].attrs["type"] == bmi.grid_type(grid_id) - assert type(grid.data_vars["node_x"].data) == np.ndarray + assert grid.data_vars["mesh"].attrs["type"] is bmi.grid_type(grid_id) + assert type(grid.data_vars["node_x"].data) is np.ndarray class BmiUnstructured(BmiPoints): @@ -106,8 +106,8 @@ def test_unstructured_grid(): assert isinstance(grid, xr.Dataset) assert grid.ndim == 2 assert grid.metadata["type"] == bmi.grid_type(grid_id) - assert grid.data_vars["mesh"].attrs["type"] == bmi.grid_type(grid_id) - assert type(grid.data_vars["node_x"].data) == np.ndarray + assert grid.data_vars["mesh"].attrs["type"] is bmi.grid_type(grid_id) + assert type(grid.data_vars["node_x"].data) is np.ndarray class BmiStructuredQuadrilateral: @@ -138,8 +138,8 @@ def test_structured_quadrilateral_grid(): assert isinstance(grid, xr.Dataset) assert grid.ndim == 2 assert grid.metadata["type"] == bmi.grid_type(grid_id) - assert grid.data_vars["mesh"].attrs["type"] == bmi.grid_type(grid_id) - assert type(grid.data_vars["node_x"].data) == np.ndarray + assert grid.data_vars["mesh"].attrs["type"] is bmi.grid_type(grid_id) + assert type(grid.data_vars["node_x"].data) is np.ndarray class BmiRectilinear: @@ -171,8 +171,8 @@ def test_rectilinear_grid(): assert isinstance(grid, xr.Dataset) assert grid.ndim == 2 assert grid.metadata["type"] == bmi.grid_type(grid_id) - assert grid.data_vars["mesh"].attrs["type"] == bmi.grid_type(grid_id) - assert type(grid.data_vars["node_x"].data) == np.ndarray + assert grid.data_vars["mesh"].attrs["type"] is bmi.grid_type(grid_id) + assert type(grid.data_vars["node_x"].data) is np.ndarray class BmiRectilinear3D: @@ -208,8 +208,8 @@ def test_rectilinear_grid_3d(): assert isinstance(grid, xr.Dataset) assert grid.ndim == 3 assert grid.metadata["type"] == bmi.grid_type(grid_id) - assert grid.data_vars["mesh"].attrs["type"] == bmi.grid_type(grid_id) - assert type(grid.data_vars["node_x"].data) == np.ndarray + assert grid.data_vars["mesh"].attrs["type"] is bmi.grid_type(grid_id) + assert type(grid.data_vars["node_x"].data) is np.ndarray assert len(grid.data_vars["node_x"].data) == np.prod(bmi.grid_shape(grid_id)) @@ -242,5 +242,5 @@ def test_uniform_rectilinear_grid(): assert isinstance(grid, xr.Dataset) assert grid.ndim == 2 assert grid.metadata["type"] == bmi.grid_type(grid_id) - assert grid.data_vars["mesh"].attrs["type"] == bmi.grid_type(grid_id) - assert type(grid.data_vars["node_x"].data) == np.ndarray + assert grid.data_vars["mesh"].attrs["type"] is bmi.grid_type(grid_id) + assert type(grid.data_vars["node_x"].data) is np.ndarray