def test_insufficient_coordinates(self):
with self.assertRaises(ValueError):
OceanSigmaFactory()
with self.assertRaises(ValueError):
OceanSigmaFactory(sigma=None, eta=self.eta, depth=self.depth)
with self.assertRaises(ValueError):
OceanSigmaFactory(sigma=self.sigma, eta=None, depth=self.depth)
with self.assertRaises(ValueError):
OceanSigmaFactory(sigma=self.sigma, eta=self.eta, depth=None)
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, ...]
# Calculate the expected result.
expected_coord = self.derive(sigma, eta, depth)
# Calculate the actual result.
factory = OceanSigmaFactory(**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, ...]
# Calculate the expected result.
expected_coord = self.derive(sigma, eta, depth)
# Calculate the actual result.
factory = OceanSigmaFactory(**self.kwargs)
coord = factory.make_coord(self.coord_dims)
self.assertEqual(expected_coord, coord)
def test_values(self):
factory = OceanSigmaFactory(**self.kwargs)
self.assertEqual(factory.dependencies, self.kwargs)
def test_depth_incompatible_units(self):
self.depth.units = Unit('km')
with self.assertRaises(ValueError):
OceanSigmaFactory(**self.kwargs)
def test_sigma_too_many_bounds(self):
self.sigma.nbounds = 4
with self.assertRaises(ValueError):
OceanSigmaFactory(**self.kwargs)
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.kwargs = dict(sigma=self.sigma, eta=self.eta, depth=self.depth)
self.factory = OceanSigmaFactory(**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.kwargs = dict(sigma=self.sigma, eta=self.eta, depth=self.depth)
self.factory = OceanSigmaFactory(**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_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 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.kwargs = dict(sigma=self.sigma, eta=self.eta, depth=self.depth)
self.factory = OceanSigmaFactory(**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.kwargs = dict(sigma=self.sigma, eta=self.eta, depth=self.depth)
self.factory = OceanSigmaFactory(**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_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_eta_incompatible_units(self):
self.eta.units = Unit("km")
with self.assertRaises(ValueError):
OceanSigmaFactory(**self.kwargs)
def test_promote_sigma_units_unknown_to_dimensionless(self):
sigma = mock.Mock(units=Unit("unknown"), nbounds=0)
self.kwargs["sigma"] = sigma
factory = OceanSigmaFactory(**self.kwargs)
self.assertEqual("1", factory.dependencies["sigma"].units)
