diff --git a/cf/data/data.py b/cf/data/data.py index 8a08ae719d..4b1ea7e945 100644 --- a/cf/data/data.py +++ b/cf/data/data.py @@ -6132,6 +6132,7 @@ def argmax(self, axis=None, unravel=False): return type(self)(a) + @daskified(_DASKIFIED_VERBOSE) def get_data(self, default=ValueError(), _units=None, _fill_value=None): """Returns the data. @@ -6144,6 +6145,7 @@ def get_data(self, default=ValueError(), _units=None, _fill_value=None): """ return self + @daskified(_DASKIFIED_VERBOSE) def get_units(self, default=ValueError()): """Return the units. @@ -6177,6 +6179,7 @@ def get_units(self, default=ValueError()): except AttributeError: return super().get_units(default=default) + @daskified(_DASKIFIED_VERBOSE) def get_calendar(self, default=ValueError()): """Return the calendar. @@ -6210,6 +6213,7 @@ def get_calendar(self, default=ValueError()): except (AttributeError, KeyError): return super().get_calendar(default=default) + @daskified(_DASKIFIED_VERBOSE) def set_calendar(self, calendar): """Set the calendar. @@ -6239,6 +6243,7 @@ def set_calendar(self, calendar): """ self.Units = Units(self.get_units(default=None), calendar) + @daskified(_DASKIFIED_VERBOSE) def set_units(self, value): """Set the units. diff --git a/cf/test/test_Data.py b/cf/test/test_Data.py index 188388b40d..db61c729eb 100644 --- a/cf/test/test_Data.py +++ b/cf/test/test_Data.py @@ -3961,6 +3961,10 @@ def test_Data_fits_in_memory(self): d = cf.Data.empty((size,), dtype=float) self.assertFalse(d.fits_in_memory()) + def test_Data_get_data(self): + d = cf.Data(9) + self.assertIs(d, d.get_data()) + if __name__ == "__main__": print("Run date:", datetime.datetime.now())