def test_uniform_bathy():
""" Test that ocean can be initialized with uniform bathymetry"""
#o = Ocean(default=False, cache=False, load_bathymetry=-500.5)
o = Ocean(load_bathymetry=500.5, **bounds)
assert o.bathy(test_lat, test_lon) == 500.5
assert o.temp(test_lat, test_lon, test_depth) == 0
def test_interp_chs_bathy():
""" Test that we can interpolate bathymetry data
obtained from a CHS file"""
o = Ocean(load_bathymetry='chs',
south=43.1, west=-59.8, north=43.8, east=-59.2,
top=0, bottom=0, start=default_val['start'], end=default_val['end'])
b = o.bathy(lat=1, lon=2)
assert isinstance(b, float)
b = o.bathy(lat=[43.2,43.7], lon=[-59.3, -59.4])
assert len(b) == 2
def test_chs_bathy():
""" Test that ocean can be initialized with bathymetry data
from a CHS file with automatic fetching enabled"""
bound_args = bounds.copy()
bound_args['south'], bound_args['west'], bound_args['north'], bound_args['east'] = 43.1, -59.8, 43.8, -59.2
o = Ocean(fetch=8, load_bathymetry='chs', **bound_args)
test_lat = [43.4, 43.5]
test_lon = [-59.6, -59.5]
bathy = o.bathy(test_lat, test_lon)
assert len(bathy) > 0 #check that some data was retrieved
assert 43.1 <= np.min(test_lat) and np.max(test_lat) <= 43.8 #check that lats are within limits
assert -59.8 <= np.min(test_lon) and np.max(test_lon) <= -59.2 #check that lons are within limits
# check that all nodes have meaningful bathymetry values
assert np.all(bathy < 10000)
assert np.all(bathy > -15000)
assert o.interps['bathy'].origin == o.origin
def test_source_level():
ok = {'load_bathymetry': 10000, 'load_wind_uv': 5.14}
o = Ocean(**ok)
sl = source_level(freq=10, x=0, y=0, area=1, ocean=o, sl_func=kewley_sl_func)
assert sl == 44.0
sl = source_level(freq=100, x=[0,100], y=[0,100], area=[1,2], ocean=o, sl_func=kewley_sl_func)
assert sl[0] == 42.5
assert sl[1] == sl[0] + 10*np.log10(2)
def test_array_bathy():
""" Test that ocean can be initialized with bathymetry data
from arrays"""
bathy = np.array([[-100., -200.],
[-100., -200.]])
lats = np.array([44.5, 44.7])
lons = np.array([-60.1, -59.5])
# note that fetching does nothing when supplying raw array data
o = Ocean(load_bathymetry=(bathy, lats, lons), fetch=True, **bounds)
la = lats
lo = lons
b = o.bathy(lat=lats, lon=lons)
assert np.all(b == bathy)
assert np.all(la == lats)
assert np.all(lo == lons)
res = o.bathy(lat=44.5, lon=-60.1)
assert res == -100
res = o.bathy(lat=44.5, lon=-59.8)
assert pytest.approx(res == -150., abs=1e-6)
def test_null_ocean():
""" Test that ocean is initialized with all variables set to
null (0) when default=False"""
#o = Ocean(default=False, cache=False)
# changed to make ocean null by default
o = Ocean(**bounds)
assert o.bathy(test_lat, test_lon) == 0
assert o.temp(test_lat, test_lon, test_depth) == 0
assert o.salinity(test_lat, test_lon, test_depth) == 0
assert o.wavedir(test_lat, test_lon) == 0
assert o.waveheight(test_lat, test_lon) == 0
assert o.waveperiod(test_lat, test_lon) == 0
assert o.wind_uv(test_lat, test_lon) == 0
assert o.origin == (45, -63.5)
assert o.boundaries == bounds
def test_small_full_ocean():
""" test that the ocean can be initialized for a very small region """
bounds = dict(
start=datetime(2015, 1, 9), end=datetime(2015, 1, 9, 3),
south=44.2, west=-64.4,
north=44.21, east=-64.39,
top=0, bottom=1
)
try:
o = Ocean(load_bathymetry='chs', load_temp='hycom', load_salinity='hycom',
load_wavedir='era5', load_waveheight='wwiii', load_waveperiod='era5',
load_wind_uv='wwiii', **bounds)
except AssertionError as err:
# this is intended behaviour
logging.info('CAUGHT EXCEPTION: ' + str(err))
pass
except Exception as err:
raise err
def test_interp_hycom_temp_chs_bathy():
""" Test that ocean can be initialized with temperature data
from HYCOM with automatic fetching enabled and using
start/end args.
"""
kwargs = dict(
south=43.1, west=-59.8,
north=43.8, east=-59.2,
top=-100, bottom=3000,
start=datetime(2015,1,1),
end=datetime(2015,1,2)
)
o = Ocean(#fetch=True, #cache=False,
load_temp='hycom',
load_bathymetry='chs',
**kwargs
)
lats = [43.4, 43.5]
lons = [-59.6, -59.5]
depths = [200, 300]
#(temp,lats,lons,depths) = o.temp()
temp = o.temp(lats, lons, depths)
assert 43.1 <= np.min(lats) and np.max(lats) <= 43.8 #check that lats are within limits
assert -59.8 <= np.min(lons) and np.max(lons) <= -59.2 #check that lons are within limits
assert 3000 >= np.max(depths) and np.min(depths) >= -100 #check that depths are within limits
assert len(temp) > 0 #check that some data was retrieved
assert len(depths) > 1 #check that more than one depth was fetched
assert np.all(np.abs(temp - temp.astype(int) > 1e-12)) # check that temperatures are not integers
# interpolate temperature at seafloor
seafloor_depth = o.bathy(lats, lons)
temp = o.temp(lats, lons, seafloor_depth)
assert np.all(np.logical_and(temp > -5, temp < 105)) # check sensible temperature at bottom
# load hycom temperature data
temp, lat, lon, epoch, depth = hycom().load_temp(**kwargs)
# select a location close to the center of the region
df = pd.DataFrame({'temp': temp, 'lat': lat, 'lon': lon, 'epoch': epoch, 'depth': depth})
lat_close = df.lat.values[np.argmin(np.abs(df.lat.values - 43.45))]
lon_close = df.lon.values[np.argmin(np.abs(df.lon.values - 59.5))]
df = df[(df.lat == lat_close) & (df.lon == lon_close)]
# query temps from ocean interpolator
depths = np.unique(df.depth.values)
temp_ocean = o.temp(lat_close, lon_close, depths, grid=True)
df = df.set_index(['depth','epoch'])
df = df.groupby(level=[0]).mean()
# check that interpolated temps and original temps agree
assert np.all(temp_ocean == df.temp.values)
def test_sound_speed_from_uniform_data():
# environment data provider
o = Ocean(load_bathymetry=1000, load_temp=4, load_salinity=3)
# instance of sound speed class
_ = SoundSpeed(o, num_depths=50, rel_err=None)
def test_interp_gebco_bathy():
o = Ocean(load_bathymetry='gebco',
south=43.1, west=-59.8, north=43.8, east=-59.2,
top=0, bottom=0, start=default_val['start'], end=default_val['end'])
def test_uniform_bathy_deriv():
""" Test that uniform bathy has derivative zero"""
#o = Ocean(default=False, cache=False, load_bathymetry=-500.5)
o = Ocean(load_bathymetry=-500.5, **bounds)
assert o.bathy_deriv(lat=1,lon=17,axis='lon') == 0
def test_interp_uniform_temp():
""" Test that we can interpolate a uniform ocean temperature
on any set of coordinates"""
#o = Ocean(default=False, cache=False, load_temp=16.1)
o = Ocean(load_temp=16.1, **bounds)
assert o.temp(lat=41.2, lon=-66.0, depth=-33.0) == 16.1
def test_wind_water_uv():
o = Ocean(load_water_u='hycom', load_water_v='hycom', load_water_uv='hycom',
load_wind_u='era5', load_wind_v='era5', load_wind_uv='era5',
**bounds)