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_zlev_same_boundedness(self):
self.zlev.nbounds = 2
with self.assertRaises(ValueError):
OceanSigmaZFactory(**self.kwargs)
def test_eta_incompatible_units(self):
self.eta.units = Unit('km')
with self.assertRaises(ValueError):
OceanSigmaZFactory(**self.kwargs)
