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)