def test_building_cube(self): """Test that a cube is built as expected.""" plugin = SpotExtraction() spot_values = np.array([0, 0, 12, 12]) result = plugin.build_diagnostic_cube( self.neighbour_cube, self.diagnostic_cube_xy, spot_values ) self.assertArrayEqual(result.coord("latitude").points, self.latitudes) self.assertArrayEqual(result.coord("longitude").points, self.longitudes) self.assertArrayEqual(result.data, spot_values)
def test_building_cube(self): """Test that a cube is built as expected.""" plugin = SpotExtraction() spot_values = np.array([0, 0, 12, 12]) result = plugin.build_diagnostic_cube( self.neighbour_cube, self.diagnostic_cube_2d_time, spot_values, unique_site_id=self.unique_site_id, unique_site_id_key=self.unique_site_id_key, auxiliary_coords=[self.expected_spot_time_coord], ) self.assertArrayEqual(result.coord("latitude").points, self.latitudes) self.assertArrayEqual( result.coord("longitude").points, self.longitudes) self.assertArrayEqual(result.coord("altitude").points, self.altitudes) self.assertArrayEqual(result.coord("wmo_id").points, self.wmo_ids) self.assertArrayEqual( result.coord(self.unique_site_id_key).points, self.unique_site_id) self.assertArrayEqual( result.coord("time").points, self.expected_spot_time_coord.points) self.assertArrayEqual(result.data, spot_values)