class Test__assert_axioms(tests.IrisTest): def setUp(self): self.real_array = np.array(0) self.lazy_array = as_lazy_data(self.real_array) self.dm = DataManager(self.real_array) def test_array_none(self): self.dm._real_array = None emsg = 'Unexpected data state, got no lazy and no real data' with self.assertRaisesRegexp(AssertionError, emsg): self.dm._assert_axioms() def test_array_all(self): self.dm._lazy_array = self.lazy_array emsg = 'Unexpected data state, got lazy and real data' with self.assertRaisesRegexp(AssertionError, emsg): self.dm._assert_axioms()
class Test__assert_axioms(tests.IrisTest): def setUp(self): self.real_array = np.array(0) self.lazy_array = as_lazy_data(self.real_array) self.dm = DataManager(self.real_array) def test_array_none(self): self.dm._real_array = None emsg = "Unexpected data state, got no lazy and no real data" with self.assertRaisesRegex(AssertionError, emsg): self.dm._assert_axioms() def test_array_all(self): self.dm._lazy_array = self.lazy_array emsg = "Unexpected data state, got lazy and real data" with self.assertRaisesRegex(AssertionError, emsg): self.dm._assert_axioms()
class Test__assert_axioms(tests.IrisTest): def setUp(self): self.real_array = np.array(0) self.lazy_array = as_lazy_data(self.real_array) self.dm = DataManager(self.real_array) def test_array_none(self): self.dm._real_array = None emsg = 'Unexpected data state, got no lazy and no real data' with self.assertRaisesRegexp(AssertionError, emsg): self.dm._assert_axioms() def test_array_all(self): self.dm._lazy_array = self.lazy_array emsg = 'Unexpected data state, got lazy and real data' with self.assertRaisesRegexp(AssertionError, emsg): self.dm._assert_axioms() def test_realised_dtype(self): self.dm._realised_dtype = np.dtype('float') emsg = 'Unexpected realised dtype state, got dtype' with self.assertRaisesRegexp(AssertionError, emsg): self.dm._assert_axioms() def test_real_array_with_realised_dtype(self): self.dm._realised_dtype = np.dtype('int') emsg = ("Unexpected real data with realised dtype, got " "real data and realised dtype\('int64'\)") with self.assertRaisesRegexp(AssertionError, emsg): self.dm._assert_axioms() def test_non_float_lazy_array_with_realised_dtype(self): self.dm._real_array = None self.dm._lazy_array = self.lazy_array self.dm._realised_dtype = np.dtype('int') emsg = ("Unexpected lazy data dtype with realised dtype, got " "lazy data dtype\('int64'\) and realised " "dtype\('int64'\)") with self.assertRaisesRegexp(AssertionError, emsg): self.dm._assert_axioms()