def test_longitude_0_360(self): x = np.arange(10, 370, 20) y = np.array([50.5, 51.5]) values = np.arange(len(y) * len(x)).reshape((len(y), len(x))) latitude = DimCoord(y, standard_name='latitude', units='degrees') longitude = DimCoord(x, standard_name='longitude', units='degrees') data = make_from_cube(Cube(values, dim_coords_and_dims=[(latitude, 0), (longitude, 1)])) out_x, out_y, out_values = make_color_mesh_cells(data, self.plot_args) x_bounds = np.arange(0, 380, 20) y_bounds = np.array([50, 51, 52]) expected_x, expected_y = np.meshgrid(x_bounds, y_bounds) assert_arrays_equal(out_x, expected_x) assert_arrays_equal(out_y, expected_y) # Test that a plot doesn't fail. map = Heatmap([data], self.plot_args) map.plot()
def test_lat_lon_decreasing_no_bounds(self): x = np.array([0.5, -0.5]) y = np.array([51.5, 50.5]) values = np.array([[1, 2], [3, 4]]) latitude = DimCoord(y, standard_name='latitude', units='degrees') longitude = DimCoord(x, standard_name='longitude', units='degrees') data = make_from_cube(Cube(values, dim_coords_and_dims=[(latitude, 0), (longitude, 1)])) out_x, out_y, out_values = make_color_mesh_cells(data, self.plot_args) expected_x = np.array([[1, 0, -1], [1, 0, -1], [1, 0, -1]]) expected_y = np.array([[52, 52, 52], [51, 51, 51], [50, 50, 50]]) expected_v = np.array([[1, 2], [3, 4]]) assert_arrays_equal(out_x, expected_x) assert_arrays_equal(out_y, expected_y) assert_arrays_equal(out_values, expected_v) # Test that a plot doesn't fail. map = Heatmap([data], self.plot_args) map.plot()