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)
