def test_FieldAncillary_properties(self):
d = self.f.domain_ancillary("ncvar%a")
x = cf.FieldAncillary(source=d)
x.set_property("long_name", "qwerty")
self.assertEqual(x.get_property("long_name"), "qwerty")
self.assertEqual(x.del_property("long_name"), "qwerty")
self.assertIsNone(x.get_property("long_name", None))
self.assertIsNone(x.del_property("long_name", None))
def test_FieldAncillary_transpose(self):
a = self.f.auxiliary_coordinate("longitude")
x = cf.FieldAncillary(source=a)
self.assertEqual(x.shape, (9, 10))
y = x.transpose()
self.assertEqual(y.shape, (10, 9))
x.transpose([1, 0], inplace=True)
self.assertEqual(x.shape, (10, 9))
def test_FieldAncillary_insert_dimension(self):
d = self.f.dimension_coordinate("grid_longitude")
x = cf.FieldAncillary(source=d)
self.assertEqual(x.shape, (9, ))
y = x.insert_dimension(0)
self.assertEqual(y.shape, (1, 9))
x.insert_dimension(-1, inplace=True)
self.assertEqual(x.shape, (9, 1))
def test_FieldAncillary_transpose(self):
f = cf.read(self.filename)[0]
a = f.auxiliary_coordinates("longitude").value()
x = cf.FieldAncillary(source=a)
self.assertEqual(x.shape, (9, 10))
y = x.transpose()
self.assertEqual(y.shape, (10, 9))
x.transpose([1, 0], inplace=True)
self.assertEqual(x.shape, (10, 9))
def test_FieldAncillary_insert_dimension(self):
f = cf.read(self.filename)[0]
d = f.dimension_coordinates("grid_longitude").value()
x = cf.FieldAncillary(source=d)
self.assertEqual(x.shape, (9, ))
y = x.insert_dimension(0)
self.assertEqual(y.shape, (1, 9))
x.insert_dimension(-1, inplace=True)
self.assertEqual(x.shape, (9, 1))
def test_FieldAncillary_squeeze(self):
a = self.f.auxiliary_coordinate("longitude")
x = cf.FieldAncillary(source=a)
x.insert_dimension(1, inplace=True)
x.insert_dimension(0, inplace=True)
self.assertEqual(x.shape, (1, 9, 1, 10))
y = x.squeeze()
self.assertEqual(y.shape, (9, 10))
x.squeeze(2, inplace=True)
self.assertEqual(x.shape, (1, 9, 10))
def test_FieldAncillary_source(self):
f = cf.read(self.filename)[0]
a = f.auxiliary_coordinates("latitude").value()
cf.FieldAncillary(source=a)
def test_FieldAncillary(self):
f = cf.FieldAncillary()
_ = repr(f)
_ = str(f)
_ = f.dump(display=False)
def test_create_field(self):
# Dimension coordinates
dim1 = cf.DimensionCoordinate(
data=cf.Data(numpy.arange(10.0), "degrees"))
dim1.standard_name = "grid_latitude"
dim0 = cf.DimensionCoordinate(
data=cf.Data(numpy.arange(9.0) + 20, "degrees"))
dim0.standard_name = "grid_longitude"
dim0.data[-1] += 5
bounds = cf.Data(
numpy.array([dim0.data.array - 0.5,
dim0.data.array + 0.5]).transpose((1, 0)))
bounds[-2, 1] = 30
bounds[-1, :] = [30, 36]
dim0.set_bounds(cf.Bounds(data=bounds))
dim2 = cf.DimensionCoordinate(
data=cf.Data([1.5]), bounds=cf.Bounds(data=cf.Data([[1, 2.0]])))
dim2.standard_name = "atmosphere_hybrid_height_coordinate"
# Auxiliary coordinates
ak = cf.DomainAncillary(data=cf.Data([10.0], "m"))
ak.id = "atmosphere_hybrid_height_coordinate_ak"
bounds = cf.Bounds(data=cf.Data([[5, 15.0]], units=ak.Units))
ak.set_bounds(bounds)
bk = cf.DomainAncillary(data=cf.Data([20.0]))
bk.id = "atmosphere_hybrid_height_coordinate_bk"
bounds = cf.Bounds(data=cf.Data([[14, 26.0]]))
bk.set_bounds(bounds)
aux2 = cf.AuxiliaryCoordinate(data=cf.Data(
numpy.arange(-45, 45, dtype="int32").reshape(10, 9),
units="degree_N",
))
aux2.standard_name = "latitude"
aux3 = cf.AuxiliaryCoordinate(data=cf.Data(
numpy.arange(60, 150, dtype="int32").reshape(9, 10),
units="degreesE",
))
aux3.standard_name = "longitude"
aux4 = cf.AuxiliaryCoordinate(data=cf.Data(
numpy.array(
[
"alpha",
"beta",
"gamma",
"delta",
"epsilon",
"zeta",
"eta",
"theta",
"iota",
"kappa",
],
dtype="S",
)))
aux4.standard_name = "greek_letters"
aux4[0] = cf.masked
# Cell measures
msr0 = cf.CellMeasure(
data=cf.Data(1 + numpy.arange(90.0).reshape(9, 10) * 1234, "km 2"))
msr0.measure = "area"
# Data
data = cf.Data(numpy.arange(90.0).reshape(10, 9), "m s-1")
properties = {"standard_name": "eastward_wind"}
f = cf.Field(properties=properties)
axisX = f.set_construct(cf.DomainAxis(9))
axisY = f.set_construct(cf.DomainAxis(10))
axisZ = f.set_construct(cf.DomainAxis(1))
f.set_data(data)
x = f.set_construct(dim0)
y = f.set_construct(dim1, axes=[axisY])
z = f.set_construct(dim2, axes=[axisZ])
lat = f.set_construct(aux2)
lon = f.set_construct(aux3, axes=["X", axisY])
f.set_construct(aux4, axes=["Y"])
ak = f.set_construct(ak, axes=["Z"])
bk = f.set_construct(bk, axes=[axisZ])
# Coordinate references
coordinate_conversion = cf.CoordinateConversion(
parameters={
"grid_mapping_name": "rotated_latitude_longitude",
"grid_north_pole_latitude": 38.0,
"grid_north_pole_longitude": 190.0,
})
ref0 = cf.CoordinateReference(
coordinate_conversion=coordinate_conversion,
coordinates=[x, y, lat, lon],
)
f.set_construct(msr0, axes=[axisX, "Y"])
f.set_construct(ref0)
orog = cf.DomainAncillary()
orog.standard_name = "surface_altitude"
orog.set_data(cf.Data(f.array * 2, "m"))
orog.transpose([1, 0], inplace=True)
orog_key = f.set_construct(orog, axes=["X", axisY])
coordinate_conversion = cf.CoordinateConversion(
parameters={
"standard_name": "atmosphere_hybrid_height_coordinate"
},
domain_ancillaries={
"orog": orog_key,
"a": ak,
"b": bk
},
)
ref1 = cf.CoordinateReference(
coordinate_conversion=coordinate_conversion, coordinates=[z])
f.set_construct(ref1)
# Field ancillary variables
g = cf.FieldAncillary()
g.set_data(f.data)
g.transpose([1, 0], inplace=True)
g.standard_name = "ancillary0"
g *= 0.01
f.set_construct(g)
g = cf.FieldAncillary()
g.set_data(f.data)
g.standard_name = "ancillary1"
g *= 0.01
f.set_construct(g)
g = cf.FieldAncillary()
g.set_data(f[0, :].data)
g.squeeze(inplace=True)
g.standard_name = "ancillary2"
g *= 0.001
f.set_construct(g)
g = cf.FieldAncillary()
g.set_data(f[:, 0].data)
g.squeeze(inplace=True)
g.standard_name = "ancillary3"
g *= 0.001
f.set_construct(g)
f.flag_values = [1, 2, 4]
f.flag_meanings = ["a", "bb", "ccc"]
for cm in cf.CellMethod.create(
"grid_longitude: mean grid_latitude: max"):
f.set_construct(cm)
# Write the file, and read it in
cf.write(f, self.filename, verbose=0, string=True)
g = cf.read(self.filename, squeeze=True, verbose=0)[0]
self.assertTrue(g.equals(f, verbose=0),
"Field not equal to itself read back in")
x = g.dump(display=False)
x = f.dump(display=False)
print("\n**Stage 1:** The field construct is created without metadata\n")
print("\n**Stage 2:** Metadata constructs are created independently.\n")
print("\n**Stage 3:** The metadata constructs are inserted into the field\n")
p = cf.Field(properties={'standard_name': 'precipitation_flux'})
p
dc = cf.DimensionCoordinate(properties={'long_name': 'Longitude'},
data=cf.Data([0, 1, 2.]))
dc
fa = cf.FieldAncillary(
properties={'standard_name': 'precipitation_flux status_flag'},
data=cf.Data(numpy.array([0, 0, 2], dtype='int8')))
fa
p = cf.Field()
p
p.set_property('standard_name', 'precipitation_flux')
p
dc = cf.DimensionCoordinate()
dc
dc.set_property('long_name', 'Longitude')
dc.set_data(cf.Data([1, 2, 3.]))
dc
fa = cf.FieldAncillary(
data=cf.Data(numpy.array([0, 0, 2], dtype='int8')))
fa
fa.set_property('standard_name', 'precipitation_flux status_flag')
def test_FieldAncillary_source(self):
a = self.f.auxiliary_coordinate("latitude")
cf.FieldAncillary(source=a)
def test_create_field(self):
# Dimension coordinates
dim1 = cf.DimensionCoordinate(
data=cf.Data(numpy.arange(10.), 'degrees'))
dim1.standard_name = 'grid_latitude'
dim0 = cf.DimensionCoordinate(
data=cf.Data(numpy.arange(9.) + 20, 'degrees'))
dim0.standard_name = 'grid_longitude'
dim0.data[-1] += 5
bounds = cf.Data(numpy.array(
[dim0.data.array-0.5, dim0.data.array+0.5]).transpose((1, 0)))
bounds[-2, 1] = 30
bounds[-1, :] = [30, 36]
dim0.set_bounds(cf.Bounds(data=bounds))
dim2 = cf.DimensionCoordinate(
data=cf.Data([1.5]),
bounds=cf.Bounds(data=cf.Data([[1, 2.]]))
)
dim2.standard_name = 'atmosphere_hybrid_height_coordinate'
# Auxiliary coordinates
ak = cf.DomainAncillary(data=cf.Data([10.], 'm'))
ak.id = 'atmosphere_hybrid_height_coordinate_ak'
bounds = cf.Bounds(data=cf.Data([[5, 15.]], units=ak.Units))
ak.set_bounds(bounds)
bk = cf.DomainAncillary(data=cf.Data([20.]))
bk.id = 'atmosphere_hybrid_height_coordinate_bk'
bounds = cf.Bounds(data=cf.Data([[14, 26.]]))
bk.set_bounds(bounds)
aux2 = cf.AuxiliaryCoordinate(
data=cf.Data(numpy.arange(-45, 45, dtype='int32').reshape(10, 9),
units='degree_N'))
aux2.standard_name = 'latitude'
aux3 = cf.AuxiliaryCoordinate(
data=cf.Data(numpy.arange(60, 150, dtype='int32').reshape(9, 10),
units='degreesE'))
aux3.standard_name = 'longitude'
aux4 = cf.AuxiliaryCoordinate(
data=cf.Data(numpy.array(
['alpha', 'beta', 'gamma', 'delta', 'epsilon',
'zeta', 'eta', 'theta', 'iota', 'kappa'],
dtype='S'
))
)
aux4.standard_name = 'greek_letters'
aux4[0] = cf.masked
# Cell measures
msr0 = cf.CellMeasure(
data=cf.Data(1+numpy.arange(90.).reshape(9, 10)*1234, 'km 2'))
msr0.measure = 'area'
# Data
data = cf.Data(numpy.arange(90.).reshape(10, 9), 'm s-1')
properties = {'standard_name': 'eastward_wind'}
f = cf.Field(properties=properties)
axisX = f.set_construct(cf.DomainAxis(9))
axisY = f.set_construct(cf.DomainAxis(10))
axisZ = f.set_construct(cf.DomainAxis(1))
f.set_data(data)
x = f.set_construct(dim0)
y = f.set_construct(dim1, axes=[axisY])
z = f.set_construct(dim2, axes=[axisZ])
lat = f.set_construct(aux2)
lon = f.set_construct(aux3, axes=['X', axisY])
f.set_construct(aux4, axes=['Y'])
ak = f.set_construct(ak, axes=['Z'])
bk = f.set_construct(bk, axes=[axisZ])
# Coordinate references
coordinate_conversion = cf.CoordinateConversion(
parameters={'grid_mapping_name': 'rotated_latitude_longitude',
'grid_north_pole_latitude': 38.0,
'grid_north_pole_longitude': 190.0})
ref0 = cf.CoordinateReference(
coordinate_conversion=coordinate_conversion,
coordinates=[x, y, lat, lon]
)
f.set_construct(msr0, axes=[axisX, 'Y'])
f.set_construct(ref0)
orog = cf.DomainAncillary()
orog.standard_name = 'surface_altitude'
orog.set_data(cf.Data(f.array*2, 'm'))
orog.transpose([1, 0], inplace=True)
orog_key = f.set_construct(orog, axes=['X', axisY])
coordinate_conversion = cf.CoordinateConversion(
parameters={
'standard_name': 'atmosphere_hybrid_height_coordinate'
},
domain_ancillaries={
'orog': orog_key,
'a': ak,
'b': bk
}
)
ref1 = cf.CoordinateReference(
coordinate_conversion=coordinate_conversion, coordinates=[z])
f.set_construct(ref1)
# Field ancillary variables
g = cf.FieldAncillary()
g.set_data(f.data)
g.transpose([1, 0], inplace=True)
g.standard_name = 'ancillary0'
g *= 0.01
f.set_construct(g)
g = cf.FieldAncillary()
g.set_data(f.data)
g.standard_name = 'ancillary1'
g *= 0.01
f.set_construct(g)
g = cf.FieldAncillary()
g.set_data(f[0, :].data)
g.squeeze(inplace=True)
g.standard_name = 'ancillary2'
g *= 0.001
f.set_construct(g)
g = cf.FieldAncillary()
g.set_data(f[:, 0].data)
g.squeeze(inplace=True)
g.standard_name = 'ancillary3'
g *= 0.001
f.set_construct(g)
f.flag_values = [1, 2, 4]
f.flag_meanings = ['a', 'bb', 'ccc']
for cm in cf.CellMethod.create(
'grid_longitude: mean grid_latitude: max'):
f.set_construct(cm)
# Write the file, and read it in
cf.write(f, self.filename, verbose=0, string=True)
g = cf.read(self.filename, squeeze=True, verbose=0)[0]
self.assertTrue(g.equals(f, verbose=0),
"Field not equal to itself read back in")
x = g.dump(display=False)
x = f.dump(display=False)
