def test_depth_integration_3d(self):
"""Test to take the depth integration of a 3 layer cube."""
result = depth_integration(self.grid_3d)
expected = np.ones((2, 2)) * 250.
print(result.data, expected.data)
self.assert_array_equal(result.data, expected)
def test_depth_integration_1d(self):
"""Test to take the depth integration of a 3 layer cube."""
result = depth_integration(self.grid_3d[:, 0, 0])
expected = np.ones((1, 1)) * 250.
print(result.data, expected.data)
self.assertArrayEqual(result.data, expected)