def test_derived_points_with_bounds(self):
self.sigma.guess_bounds()
self.zlev.guess_bounds()
# Broadcast expected points given the known dimensional mapping.
sigma = self.sigma.points[..., np.newaxis, np.newaxis]
eta = self.eta.points[np.newaxis, ...]
depth = self.depth.points[np.newaxis, ...]
depth_c = self.depth_c.points
nsigma = self.nsigma.points
zlev = self.zlev.points[..., np.newaxis, np.newaxis]
# Calculate the expected coordinate with points.
expected_coord = self.derive(sigma, eta, depth, depth_c, nsigma, zlev)
# Broadcast expected bounds given the known dimensional mapping.
sigma = self.sigma.bounds.reshape(sigma.shape + (2, ))
eta = self.eta.points.reshape(eta.shape + (1, ))
depth = self.depth.points.reshape(depth.shape + (1, ))
depth_c = self.depth_c.points.reshape(depth_c.shape + (1, ))
nsigma = self.nsigma.points.reshape(nsigma.shape + (1, ))
zlev = self.zlev.bounds.reshape(zlev.shape + (2, ))
# Calculate the expected bounds.
bounds = self.derive(sigma,
eta,
depth,
depth_c,
nsigma,
zlev,
coord=False)
expected_coord.bounds = bounds
# Calculate the actual result.
factory = OceanSigmaZFactory(**self.kwargs)
coord = factory.make_coord(self.coord_dims)
self.assertEqual(expected_coord, coord)
def test_derived_points_with_bounds(self):
self.sigma.guess_bounds()
self.zlev.guess_bounds()
# Broadcast expected points given the known dimensional mapping.
sigma = self.sigma.points[..., np.newaxis, np.newaxis]
eta = self.eta.points[np.newaxis, ...]
depth = self.depth.points[np.newaxis, ...]
depth_c = self.depth_c.points
nsigma = self.nsigma.points
zlev = self.zlev.points[..., np.newaxis, np.newaxis]
# Calculate the expected coordinate with points.
expected_coord = self.derive(sigma, eta, depth, depth_c, nsigma, zlev)
# Broadcast expected bounds given the known dimensional mapping.
sigma = self.sigma.bounds.reshape(sigma.shape + (2,))
eta = self.eta.points.reshape(eta.shape + (1,))
depth = self.depth.points.reshape(depth.shape + (1,))
depth_c = self.depth_c.points.reshape(depth_c.shape + (1,))
nsigma = self.nsigma.points.reshape(nsigma.shape + (1,))
zlev = self.zlev.bounds.reshape(zlev.shape + (2,))
# Calculate the expected bounds.
bounds = self.derive(sigma, eta, depth, depth_c, nsigma, zlev,
coord=False)
expected_coord.bounds = bounds
# Calculate the actual result.
factory = OceanSigmaZFactory(**self.kwargs)
coord = factory.make_coord(self.coord_dims)
self.assertEqual(expected_coord, coord)
def setUp(self):
self.sigma = mock.Mock(units=Unit('1'), nbounds=0)
self.eta = mock.Mock(units=Unit('m'), nbounds=0)
self.depth = mock.Mock(units=Unit('m'), nbounds=0)
self.depth_c = mock.Mock(units=Unit('m'), nbounds=0, shape=(1,))
self.nsigma = mock.Mock(units=Unit('1'), nbounds=0, shape=(1,))
self.zlev = mock.Mock(units=Unit('m'), nbounds=0)
self.kwargs = dict(sigma=self.sigma, eta=self.eta,
depth=self.depth, depth_c=self.depth_c,
nsigma=self.nsigma, zlev=self.zlev)
self.factory = OceanSigmaZFactory(**self.kwargs)
def test_derived_points(self):
# Broadcast expected points given the known dimensional mapping.
sigma = self.sigma.points[..., np.newaxis, np.newaxis]
eta = self.eta.points[np.newaxis, ...]
depth = self.depth.points[np.newaxis, ...]
depth_c = self.depth_c.points
nsigma = self.nsigma.points
zlev = self.zlev.points[..., np.newaxis, np.newaxis]
# Calculate the expected result.
expected_coord = self.derive(sigma, eta, depth, depth_c, nsigma, zlev)
# Calculate the actual result.
factory = OceanSigmaZFactory(**self.kwargs)
coord = factory.make_coord(self.coord_dims)
self.assertEqual(expected_coord, coord)
def test_derived_points(self):
# Broadcast expected points given the known dimensional mapping.
sigma = self.sigma.points[..., np.newaxis, np.newaxis]
eta = self.eta.points[np.newaxis, ...]
depth = self.depth.points[np.newaxis, ...]
depth_c = self.depth_c.points
nsigma = self.nsigma.points
zlev = self.zlev.points[..., np.newaxis, np.newaxis]
# Calculate the expected result.
expected_coord = self.derive(sigma, eta, depth, depth_c, nsigma, zlev)
# Calculate the actual result.
factory = OceanSigmaZFactory(**self.kwargs)
coord = factory.make_coord(self.coord_dims)
self.assertEqual(expected_coord, coord)
def setUp(self):
self.sigma = mock.Mock(units=Unit('1'), nbounds=0)
self.eta = mock.Mock(units=Unit('m'), nbounds=0)
self.depth = mock.Mock(units=Unit('m'), nbounds=0)
self.depth_c = mock.Mock(units=Unit('m'), nbounds=0, shape=(1,))
self.nsigma = mock.Mock(units=Unit('1'), nbounds=0, shape=(1,))
self.zlev = mock.Mock(units=Unit('m'), nbounds=0)
self.kwargs = dict(sigma=self.sigma, eta=self.eta,
depth=self.depth, depth_c=self.depth_c,
nsigma=self.nsigma, zlev=self.zlev)
self.factory = OceanSigmaZFactory(**self.kwargs)
def test_insufficient_coordinates(self):
with self.assertRaises(ValueError):
OceanSigmaZFactory()
with self.assertRaises(ValueError):
OceanSigmaZFactory(
sigma=self.sigma,
eta=self.eta,
depth=self.depth,
depth_c=self.depth_c,
nsigma=self.nsigma,
zlev=None,
)
with self.assertRaises(ValueError):
OceanSigmaZFactory(
sigma=None,
eta=None,
depth=self.depth,
depth_c=self.depth_c,
nsigma=self.nsigma,
zlev=self.zlev,
)
with self.assertRaises(ValueError):
OceanSigmaZFactory(
sigma=self.sigma,
eta=None,
depth=None,
depth_c=self.depth_c,
nsigma=self.nsigma,
zlev=self.zlev,
)
with self.assertRaises(ValueError):
OceanSigmaZFactory(
sigma=self.sigma,
eta=None,
depth=self.depth,
depth_c=None,
nsigma=self.nsigma,
zlev=self.zlev,
)
with self.assertRaises(ValueError):
OceanSigmaZFactory(
sigma=self.sigma,
eta=self.eta,
depth=self.depth,
depth_c=self.depth_c,
nsigma=None,
zlev=self.zlev,
)
class Test_update(tests.IrisTest):
def setUp(self):
self.sigma = mock.Mock(units=Unit('1'), nbounds=0)
self.eta = mock.Mock(units=Unit('m'), nbounds=0)
self.depth = mock.Mock(units=Unit('m'), nbounds=0)
self.depth_c = mock.Mock(units=Unit('m'), nbounds=0, shape=(1,))
self.nsigma = mock.Mock(units=Unit('1'), nbounds=0, shape=(1,))
self.zlev = mock.Mock(units=Unit('m'), nbounds=0)
self.kwargs = dict(sigma=self.sigma, eta=self.eta,
depth=self.depth, depth_c=self.depth_c,
nsigma=self.nsigma, zlev=self.zlev)
self.factory = OceanSigmaZFactory(**self.kwargs)
def test_sigma(self):
new_sigma = mock.Mock(units=Unit('1'), nbounds=0)
self.factory.update(self.sigma, new_sigma)
self.assertIs(self.factory.sigma, new_sigma)
def test_sigma_too_many_bounds(self):
new_sigma = mock.Mock(units=Unit('1'), nbounds=4)
with self.assertRaises(ValueError):
self.factory.update(self.sigma, new_sigma)
def test_sigma_zlev_same_boundedness(self):
new_sigma = mock.Mock(units=Unit('1'), nbounds=2)
with self.assertRaises(ValueError):
self.factory.update(self.sigma, new_sigma)
def test_sigma_incompatible_units(self):
new_sigma = mock.Mock(units=Unit('Pa'), nbounds=0)
with self.assertRaises(ValueError):
self.factory.update(self.sigma, new_sigma)
def test_eta(self):
new_eta = mock.Mock(units=Unit('m'), nbounds=0)
self.factory.update(self.eta, new_eta)
self.assertIs(self.factory.eta, new_eta)
def test_eta_incompatible_units(self):
new_eta = mock.Mock(units=Unit('Pa'), nbounds=0)
with self.assertRaises(ValueError):
self.factory.update(self.eta, new_eta)
def test_depth(self):
new_depth = mock.Mock(units=Unit('m'), nbounds=0)
self.factory.update(self.depth, new_depth)
self.assertIs(self.factory.depth, new_depth)
def test_depth_incompatible_units(self):
new_depth = mock.Mock(units=Unit('Pa'), nbounds=0)
with self.assertRaises(ValueError):
self.factory.update(self.depth, new_depth)
def test_depth_c(self):
new_depth_c = mock.Mock(units=Unit('m'), nbounds=0, shape=(1,))
self.factory.update(self.depth_c, new_depth_c)
self.assertIs(self.factory.depth_c, new_depth_c)
def test_depth_c_non_scalar(self):
new_depth_c = mock.Mock(units=Unit('m'), nbounds=0, shape=(10,))
with self.assertRaises(ValueError):
self.factory.update(self.depth_c, new_depth_c)
def test_depth_c_incompatible_units(self):
new_depth_c = mock.Mock(units=Unit('Pa'), nbounds=0, shape=(1,))
with self.assertRaises(ValueError):
self.factory.update(self.depth_c, new_depth_c)
def test_nsigma(self):
new_nsigma = mock.Mock(units=Unit('1'), nbounds=0, shape=(1,))
self.factory.update(self.nsigma, new_nsigma)
self.assertIs(self.factory.nsigma, new_nsigma)
def test_nsigma_missing(self):
with self.assertRaises(ValueError):
self.factory.update(self.nsigma, None)
def test_nsigma_non_scalar(self):
new_nsigma = mock.Mock(units=Unit('1'), nbounds=0, shape=(10,))
with self.assertRaises(ValueError):
self.factory.update(self.nsigma, new_nsigma)
def test_zlev(self):
new_zlev = mock.Mock(units=Unit('m'), nbounds=0)
self.factory.update(self.zlev, new_zlev)
self.assertIs(self.factory.zlev, new_zlev)
def test_zlev_missing(self):
with self.assertRaises(ValueError):
self.factory.update(self.zlev, None)
def test_zlev_too_many_bounds(self):
new_zlev = mock.Mock(units=Unit('m'), nbounds=4)
with self.assertRaises(ValueError):
self.factory.update(self.zlev, new_zlev)
def test_zlev_same_boundedness(self):
new_zlev = mock.Mock(units=Unit('m'), nbounds=2)
with self.assertRaises(ValueError):
self.factory.update(self.zlev, new_zlev)
def test_zlev_incompatible_units(self):
new_zlev = new_zlev = mock.Mock(units=Unit('Pa'), nbounds=0)
with self.assertRaises(ValueError):
self.factory.update(self.zlev, new_zlev)
def test_sigma_zlev_same_boundedness(self):
self.zlev.nbounds = 2
with self.assertRaises(ValueError):
OceanSigmaZFactory(**self.kwargs)
def test_zlev_too_many_bounds(self):
self.zlev.nbounds = 4
with self.assertRaises(ValueError):
OceanSigmaZFactory(**self.kwargs)
class Test_update(tests.IrisTest):
def setUp(self):
self.sigma = mock.Mock(units=Unit("1"), nbounds=0)
self.eta = mock.Mock(units=Unit("m"), nbounds=0)
self.depth = mock.Mock(units=Unit("m"), nbounds=0)
self.depth_c = mock.Mock(units=Unit("m"), nbounds=0, shape=(1, ))
self.nsigma = mock.Mock(units=Unit("1"), nbounds=0, shape=(1, ))
self.zlev = mock.Mock(units=Unit("m"), nbounds=0)
self.kwargs = dict(
sigma=self.sigma,
eta=self.eta,
depth=self.depth,
depth_c=self.depth_c,
nsigma=self.nsigma,
zlev=self.zlev,
)
self.factory = OceanSigmaZFactory(**self.kwargs)
def test_sigma(self):
new_sigma = mock.Mock(units=Unit("1"), nbounds=0)
self.factory.update(self.sigma, new_sigma)
self.assertIs(self.factory.sigma, new_sigma)
def test_sigma_too_many_bounds(self):
new_sigma = mock.Mock(units=Unit("1"), nbounds=4)
with self.assertRaises(ValueError):
self.factory.update(self.sigma, new_sigma)
def test_sigma_zlev_same_boundedness(self):
new_sigma = mock.Mock(units=Unit("1"), nbounds=2)
with self.assertRaises(ValueError):
self.factory.update(self.sigma, new_sigma)
def test_sigma_incompatible_units(self):
new_sigma = mock.Mock(units=Unit("Pa"), nbounds=0)
with self.assertRaises(ValueError):
self.factory.update(self.sigma, new_sigma)
def test_eta(self):
new_eta = mock.Mock(units=Unit("m"), nbounds=0)
self.factory.update(self.eta, new_eta)
self.assertIs(self.factory.eta, new_eta)
def test_eta_incompatible_units(self):
new_eta = mock.Mock(units=Unit("Pa"), nbounds=0)
with self.assertRaises(ValueError):
self.factory.update(self.eta, new_eta)
def test_depth(self):
new_depth = mock.Mock(units=Unit("m"), nbounds=0)
self.factory.update(self.depth, new_depth)
self.assertIs(self.factory.depth, new_depth)
def test_depth_incompatible_units(self):
new_depth = mock.Mock(units=Unit("Pa"), nbounds=0)
with self.assertRaises(ValueError):
self.factory.update(self.depth, new_depth)
def test_depth_c(self):
new_depth_c = mock.Mock(units=Unit("m"), nbounds=0, shape=(1, ))
self.factory.update(self.depth_c, new_depth_c)
self.assertIs(self.factory.depth_c, new_depth_c)
def test_depth_c_non_scalar(self):
new_depth_c = mock.Mock(units=Unit("m"), nbounds=0, shape=(10, ))
with self.assertRaises(ValueError):
self.factory.update(self.depth_c, new_depth_c)
def test_depth_c_incompatible_units(self):
new_depth_c = mock.Mock(units=Unit("Pa"), nbounds=0, shape=(1, ))
with self.assertRaises(ValueError):
self.factory.update(self.depth_c, new_depth_c)
def test_nsigma(self):
new_nsigma = mock.Mock(units=Unit("1"), nbounds=0, shape=(1, ))
self.factory.update(self.nsigma, new_nsigma)
self.assertIs(self.factory.nsigma, new_nsigma)
def test_nsigma_missing(self):
with self.assertRaises(ValueError):
self.factory.update(self.nsigma, None)
def test_nsigma_non_scalar(self):
new_nsigma = mock.Mock(units=Unit("1"), nbounds=0, shape=(10, ))
with self.assertRaises(ValueError):
self.factory.update(self.nsigma, new_nsigma)
def test_zlev(self):
new_zlev = mock.Mock(units=Unit("m"), nbounds=0)
self.factory.update(self.zlev, new_zlev)
self.assertIs(self.factory.zlev, new_zlev)
def test_zlev_missing(self):
with self.assertRaises(ValueError):
self.factory.update(self.zlev, None)
def test_zlev_too_many_bounds(self):
new_zlev = mock.Mock(units=Unit("m"), nbounds=4)
with self.assertRaises(ValueError):
self.factory.update(self.zlev, new_zlev)
def test_zlev_same_boundedness(self):
new_zlev = mock.Mock(units=Unit("m"), nbounds=2)
with self.assertRaises(ValueError):
self.factory.update(self.zlev, new_zlev)
def test_zlev_incompatible_units(self):
new_zlev = new_zlev = mock.Mock(units=Unit("Pa"), nbounds=0)
with self.assertRaises(ValueError):
self.factory.update(self.zlev, new_zlev)
def test_eta_incompatible_units(self):
self.eta.units = Unit('km')
with self.assertRaises(ValueError):
OceanSigmaZFactory(**self.kwargs)
def test_depth_c_non_scalar(self):
self.depth_c.shape = (2, )
with self.assertRaises(ValueError):
OceanSigmaZFactory(**self.kwargs)
def test_values(self):
factory = OceanSigmaZFactory(**self.kwargs)
self.assertEqual(factory.dependencies, self.kwargs)
def test_promote_sigma_units_unknown_to_dimensionless(self):
sigma = mock.Mock(units=Unit("unknown"), nbounds=0)
self.kwargs["sigma"] = sigma
factory = OceanSigmaZFactory(**self.kwargs)
self.assertEqual("1", factory.dependencies["sigma"].units)
def test_depth_incompatible_units(self):
self.depth.units = Unit("km")
with self.assertRaises(ValueError):
OceanSigmaZFactory(**self.kwargs)
def test_zlev_incompatible_units(self):
self.zlev.units = Unit("Pa")
with self.assertRaises(ValueError):
OceanSigmaZFactory(**self.kwargs)
def test_nsigma_non_scalar(self):
self.nsigma.shape = (4, )
with self.assertRaises(ValueError):
OceanSigmaZFactory(**self.kwargs)
