def test_basemap_global_matches(test_data): reader_global = reader_global_landmask.Reader() reader_basemap = reader_basemap_landmask.Reader(llcrnrlon=4, llcrnrlat=59, urcrnrlon=18, urcrnrlat=68.1, resolution='i', projection='merc') reader_nordic = reader_ROMS_native.Reader( test_data + '2Feb2016_Nordic_sigma_3d/Nordic-4km_SLEVELS_avg_00_subset2Feb2016.nc') land = (np.array([15.]), np.array([65.6])) ocean = (np.array([5.]), np.array([65.6])) # basemap ob = OceanDrift(loglevel=00) ob.add_reader([reader_nordic, reader_basemap]) en, en_prof, missing = ob.get_environment(['land_binary_mask'], reader_nordic.start_time, land[0], land[1], np.array([0]), None) assert en.land_binary_mask == np.array([True]) en, en_prof, missing = ob.get_environment(['land_binary_mask'], reader_nordic.start_time, ocean[0], ocean[1], np.array([0]), None) assert en.land_binary_mask == np.array([False]) assert len(ob.readers) == 2 # global landmask oc = OceanDrift(loglevel=00) oc.add_reader([reader_nordic, reader_global]) en, en_prof, missing = oc.get_environment(['land_binary_mask'], reader_nordic.start_time, land[0], land[1], np.array([0]), None) assert en.land_binary_mask == np.array([True]) en, en_prof, missing = oc.get_environment(['land_binary_mask'], reader_nordic.start_time, ocean[0], ocean[1], np.array([0]), None) assert en.land_binary_mask == np.array([False]) assert len( oc.readers) == 2 # make sure opendrift doesn't add default basemap
def test_global_array(test_data): reader_global = reader_global_landmask.Reader() reader_nordic = reader_ROMS_native.Reader(test_data + '2Feb2016_Nordic_sigma_3d/Nordic-4km_SLEVELS_avg_00_subset2Feb2016.nc') lon = np.array([15., 5.]) lat = np.array([65.6, 65.6]) # global oc = OceanDrift(loglevel = 00) oc.add_reader ([reader_nordic, reader_global]) en, en_prof, missing = oc.get_environment (['land_binary_mask'], reader_nordic.start_time, lon, lat, np.array([0, 0]), None) np.testing.assert_array_equal(en.land_binary_mask, np.array([True, False])) assert len(oc.readers) == 2 # make sure opendrift doesn't add default basemap
def test_global_array(test_data): shpfilename = shpreader.natural_earth(resolution='110m', category='cultural', name='admin_0_countries') reader_landmask = reader_shape.Reader.from_shpfiles(shpfilename) reader_nordic = reader_ROMS_native.Reader(test_data + '2Feb2016_Nordic_sigma_3d/Nordic-4km_SLEVELS_avg_00_subset2Feb2016.nc') lon = np.array([15., 5.]) lat = np.array([65.6, 65.6]) # global oc = OceanDrift(loglevel = 00) oc.add_reader ([reader_nordic, reader_landmask]) en, en_prof, missing = oc.get_environment (['land_binary_mask'], reader_nordic.start_time, lon, lat, np.array([0, 0]), None) np.testing.assert_array_equal(en.land_binary_mask, np.array([True, False])) assert len(oc.readers) == 2 # make sure opendrift doesn't add default basemap