def test_projection_object(self): from yt.utilities.on_demand_imports import _cartopy as cartopy shortlist = ['Orthographic', 'PlateCarree', 'Mollweide'] for transform in shortlist: projection = get_mpl_transform(transform) proj_type = type(projection) self.slc = yt.SlicePlot(self.ds, "altitude", "Density") self.slc.set_mpl_projection(projection) assert isinstance(self.slc._projection, proj_type) assert isinstance(self.slc._transform, cartopy.crs.PlateCarree) assert isinstance(self.slc.plots['Density'].axes.projection, proj_type)
def test_geo_projections(self): from yt.utilities.on_demand_imports import _cartopy as cartopy self.slc = yt.SlicePlot(self.ds, "altitude", "Density") for transform in transform_list: if transform == 'UTM': # this requires special arguments so let's skip it continue if transform == 'OSNI': # avoid crashes, see https://github.com/SciTools/cartopy/issues/1177 continue self.slc.set_mpl_projection(transform) proj_type = type(get_mpl_transform(transform)) assert isinstance(self.slc._projection, proj_type) assert isinstance(self.slc._transform, cartopy.crs.PlateCarree) assert isinstance(self.slc.plots['Density'].axes.projection, proj_type)
def test_projection_object(self): from yt.utilities.on_demand_imports import _cartopy as cartopy shortlist = ["Orthographic", "PlateCarree", "Mollweide"] for transform in shortlist: projection = get_mpl_transform(transform) proj_type = type(projection) self.slc = yt.SlicePlot(self.ds, "altitude", ("stream", "Density"), origin="native") self.slc.set_mpl_projection(projection) assert isinstance(self.slc._projection, proj_type) assert isinstance(self.slc._transform, cartopy.crs.PlateCarree) assert isinstance( self.slc.plots[("stream", "Density")].axes.projection, proj_type)
def test_nondefault_transform(self): from yt.utilities.on_demand_imports import _cartopy as cartopy axis = "altitude" self.ds.coordinates.data_transform[axis] = "Miller" self.slc = yt.SlicePlot(self.ds, axis, "Density") shortlist = ['Orthographic', 'PlateCarree', 'Mollweide'] for transform in shortlist: self.slc.set_mpl_projection(transform) proj_type = type(get_mpl_transform(transform)) assert isinstance(self.slc._projection, proj_type) assert isinstance(self.slc._transform, cartopy.crs.Miller) assert self.ds.coordinates.data_projection[axis] == "Mollweide" assert self.ds.coordinates.data_transform[axis] == "Miller" assert isinstance(self.slc.plots['Density'].axes.projection, proj_type)