def test_lazy_data(self): # Getting lazy data realises them. ancill_var = AncillaryVariable(self.data_lazy) self.assertTrue(ancill_var.has_lazy_data()) result = ancill_var.data self.assertFalse(ancill_var.has_lazy_data()) self.assertEqualRealArraysAndDtypes(result, self.data_real)
def test_preserves_lazy(self): test_data = np.array([[11.1, 12.2, 13.3], [21.4, 22.5, 23.6]]) lazy_data = as_lazy_data(test_data) ancill_var = AncillaryVariable(data=lazy_data, units="m") ancill_var.convert_units("ft") self.assertTrue(ancill_var.has_lazy_data()) test_data_ft = Unit("m").convert(test_data, "ft") self.assertArrayAllClose(ancill_var.data, test_data_ft)
def test_masked_data_real(self): data = self.masked_data_real self.assertTrue(ma.isMaskedArray(data)) self.assertTrue(ma.count_masked(data)) ancill_var = AncillaryVariable(data) self.assertFalse(ancill_var.has_lazy_data()) self.assertTrue(ma.isMaskedArray(ancill_var.data)) self.assertTrue(ma.count_masked(ancill_var.data))
def test_masked_data_lazy(self): data = self.masked_data_lazy computed = data.compute() self.assertTrue(ma.isMaskedArray(computed)) self.assertTrue(ma.count_masked(computed)) ancill_var = AncillaryVariable(data) self.assertTrue(ancill_var.has_lazy_data()) self.assertTrue(ma.isMaskedArray(ancill_var.data)) self.assertTrue(ma.count_masked(ancill_var.data))
def test_lazy_core_realise(self): ancill_var = AncillaryVariable(self.data_lazy) ancill_var.data result = ancill_var.has_lazy_data() self.assertFalse(result)
def test_lazy_core(self): ancill_var = AncillaryVariable(self.data_lazy) result = ancill_var.has_lazy_data() self.assertTrue(result)