def test_set_variable(self):
var = Variable(name='test', value=[1, 2], dimensions='two')
dmap = DimensionMap()
dmap.set_variable(DMK.X, var, dimension='not_two')
actual = dmap.get_dimension(DMK.X)
desired = ['not_two']
self.assertEqual(actual, desired)
# Test setting with a variable and bounds.
var = Variable(name='center',
value=[1, 2, 3],
dtype=float,
dimensions='one')
var.set_extrapolated_bounds('bounds_data', 'bounds')
dmap = DimensionMap()
dmap.set_variable(DMK.Y, var)
self.assertEqual(dmap.get_bounds(DMK.Y), var.bounds.name)
new_var = Variable(name='new_center',
value=[1, 2, 3],
dtype=float,
dimensions='one')
dmap.set_variable(DMK.Y, new_var)
self.assertIsNone(dmap.get_bounds(DMK.Y))
# Test a dimension position argument is needed if the variable has more than one dimension.
var = Variable(name='two_dims',
value=np.zeros((4, 5)),
dimensions=['one', 'two'])
dmap = DimensionMap()
with self.assertRaises(DimensionMapError):
dmap.set_variable(DMK.X, var)
dmap.set_variable(DMK.X, var, pos=1)
self.assertEqual(dmap.get_dimension(DMK.X), ['two'])
# Test a scalar dimension.
var = Variable(name='scalar_dimension', dimensions=[])
dmap = DimensionMap()
dmap.set_variable(DMK.LEVEL, var)
self.assertEqual(dmap.get_variable(DMK.LEVEL), 'scalar_dimension')
self.assertIsNone(dmap.get_bounds(DMK.LEVEL))
self.assertEqual(dmap.get_dimension(DMK.LEVEL), [])
# Test dimensionless variable.
var = Variable(name='two_dims',
value=np.zeros((4, 5)),
dimensions=['one', 'two'])
dmap = DimensionMap()
dmap.set_variable(DMK.X, var, dimensionless=True)
self.assertEqual(dmap.get_dimension(DMK.X), [])
def test_set_variable(self):
var = Variable(name='test', value=[1, 2], dimensions='two')
dmap = DimensionMap()
dmap.set_variable(DMK.X, var, dimension='not_two')
actual = dmap.get_dimension(DMK.X)
desired = ['not_two']
self.assertEqual(actual, desired)
# Test setting with a variable and bounds.
var = Variable(name='center', value=[1, 2, 3], dtype=float, dimensions='one')
var.set_extrapolated_bounds('bounds_data', 'bounds')
dmap = DimensionMap()
dmap.set_variable(DMK.Y, var)
self.assertEqual(dmap.get_bounds(DMK.Y), var.bounds.name)
new_var = Variable(name='new_center', value=[1, 2, 3], dtype=float, dimensions='one')
dmap.set_variable(DMK.Y, new_var)
self.assertIsNone(dmap.get_bounds(DMK.Y))
# Test a dimension position argument is needed if the variable has more than one dimension.
var = Variable(name='two_dims', value=np.zeros((4, 5)), dimensions=['one', 'two'])
dmap = DimensionMap()
with self.assertRaises(DimensionMapError):
dmap.set_variable(DMK.X, var)
dmap.set_variable(DMK.X, var, pos=1)
self.assertEqual(dmap.get_dimension(DMK.X), ['two'])
# Test a scalar dimension.
var = Variable(name='scalar_dimension', dimensions=[])
dmap = DimensionMap()
dmap.set_variable(DMK.LEVEL, var)
self.assertEqual(dmap.get_variable(DMK.LEVEL), 'scalar_dimension')
self.assertIsNone(dmap.get_bounds(DMK.LEVEL))
self.assertEqual(dmap.get_dimension(DMK.LEVEL), [])
# Test dimensionless variable.
var = Variable(name='two_dims', value=np.zeros((4, 5)), dimensions=['one', 'two'])
dmap = DimensionMap()
dmap.set_variable(DMK.X, var, dimensionless=True)
self.assertEqual(dmap.get_dimension(DMK.X), [])
def test(self):
dmap = DimensionMap()
_ = dmap.get_variable('x')
_ = dmap.get_dimension('x')
_ = dmap.get_attrs('x')
_ = dmap.get_bounds('x')
dmap.set_variable('y', 'latitude', dimension='ache', bounds='lat_bounds')
_ = dmap.get_crs()
dmap.set_crs('latitude_longitude')
desired = {'crs': {'variable': 'latitude_longitude'},
'x': {'attrs': {}, 'bounds': None, 'dimension': [], 'variable': None},
'y': {'attrs': {},
'bounds': 'lat_bounds',
'dimension': ['ache'],
'variable': 'latitude'}}
self.assertDictEqual(dmap._storage, desired)
def test(self):
dmap = DimensionMap()
_ = dmap.get_variable('x')
_ = dmap.get_dimension('x')
_ = dmap.get_attrs('x')
_ = dmap.get_bounds('x')
dmap.set_variable('y',
'latitude',
dimension='ache',
bounds='lat_bounds')
_ = dmap.get_crs()
dmap.set_crs('latitude_longitude')
desired = {
'crs': {
'variable': 'latitude_longitude'
},
'x': {
'attrs': {},
'bounds': None,
'dimension': [],
'variable': None
},
'y': {
'attrs': {
'axis': 'Y'
},
'bounds': 'lat_bounds',
'dimension': ['ache'],
'variable': 'latitude'
}
}
self.assertDictEqual(dmap._storage, desired)