def test_GATHERING_create(self):
if self.test_only and inspect.stack()[0][3] not in self.test_only:
return
# Define the gathered values
gathered_array = numpy.array([[280, 282.5, 281], [279, 278, 277.5]],
dtype='float32')
# Define the list array values
list_array = [1, 4, 5]
# Initialise the list variable
list_variable = cfdm.List(data=cfdm.Data(list_array))
# Initialise the gathered array object
array = cfdm.GatheredArray(
compressed_array=cfdm.Data(gathered_array),
compressed_dimension=1,
shape=(2, 3, 2), size=12, ndim=3,
list_variable=list_variable
)
# Create the field construct with the domain axes and the
# gathered array
tas = cfdm.Field(properties={'standard_name': 'air_temperature',
'units': 'K'})
# Create the domain axis constructs for the uncompressed array
T = tas.set_construct(cfdm.DomainAxis(2))
Y = tas.set_construct(cfdm.DomainAxis(3))
X = tas.set_construct(cfdm.DomainAxis(2))
# Set the data for the field
tas.set_data(cfdm.Data(array), axes=[T, Y, X])
self.assertTrue((tas.data.array == numpy.ma.masked_array(
data=[[[1, 280.0],
[1, 1],
[282.5, 281.0]],
[[1, 279.0],
[1, 1],
[278.0, 277.5]]],
mask=[[[True, False],
[True, True],
[False, False]],
[[True, False],
[True, True],
[False, False]]],
fill_value=1e+20,
dtype='float32')).all())
self.assertEqual(tas.data.get_compression_type(), 'gathered')
self.assertTrue((tas.data.compressed_array == numpy.array(
[[280., 282.5, 281.],
[279., 278., 277.5]], dtype='float32')).all())
self.assertTrue((tas.data.get_list().data.array == numpy.array(
[1, 4, 5])).all())
def test_GATHERING_create(self):
"""Test the creation of a construct with a gathered array."""
# Define the gathered values
gathered_array = numpy.array([[280, 282.5, 281], [279, 278, 277.5]],
dtype="float32")
# Define the list array values
list_array = [1, 4, 5]
# Initialise the list variable
list_variable = cfdm.List(data=cfdm.Data(list_array))
# Initialise the gathered array object
array = cfdm.GatheredArray(
compressed_array=cfdm.Data(gathered_array),
compressed_dimension=1,
shape=(2, 3, 2),
size=12,
ndim=3,
list_variable=list_variable,
)
# Create the field construct with the domain axes and the
# gathered array
tas = cfdm.Field(properties={
"standard_name": "air_temperature",
"units": "K"
})
# Create the domain axis constructs for the uncompressed array
T = tas.set_construct(cfdm.DomainAxis(2))
Y = tas.set_construct(cfdm.DomainAxis(3))
X = tas.set_construct(cfdm.DomainAxis(2))
# Set the data for the field
tas.set_data(cfdm.Data(array), axes=[T, Y, X])
self.assertTrue((tas.data.array == numpy.ma.masked_array(
data=[
[[1, 280.0], [1, 1], [282.5, 281.0]],
[[1, 279.0], [1, 1], [278.0, 277.5]],
],
mask=[
[[True, False], [True, True], [False, False]],
[[True, False], [True, True], [False, False]],
],
fill_value=1e20,
dtype="float32",
)).all())
self.assertEqual(tas.data.get_compression_type(), "gathered")
self.assertTrue((tas.data.compressed_array == numpy.array(
[[280.0, 282.5, 281.0], [279.0, 278.0, 277.5]],
dtype="float32",
)).all())
self.assertTrue(
(tas.data.get_list().data.array == numpy.array([1, 4, 5])).all())
import cfdm
# Define the gathered values
gathered_array = cfdm.Data([[2, 1, 3], [4, 0, 5]])
# Define the list array values
list_array = [1, 4, 5]
# Create the list variable
list_variable = cfdm.List(data=cfdm.Data(list_array))
# Create the gathered array object, specifying the uncompressed
# shape
array = cfdm.GatheredArray(
compressed_array=gathered_array,
compressed_dimension=1,
shape=(2, 3, 2), size=12, ndim=3,
list_variable=list_variable)
# Create the field construct with the domain axes and the gathered
# array
P = cfdm.Field(properties={'standard_name': 'precipitation_flux',
'units': 'kg m-2 s-1'})
# Create the domain axis constructs for the uncompressed array
T = P.set_construct(cfdm.DomainAxis(2))
Y = P.set_construct(cfdm.DomainAxis(3))
X = P.set_construct(cfdm.DomainAxis(2))
# Set the data for the field
P.set_data(cfdm.Data(array), axes=[T, Y, X])