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)
def test_get_group(self):
dmap = DimensionMap()
dmap.set_crs('who')
actual = dmap.get_group(None)
self.assertEqual(dmap.as_dict(), actual.as_dict())
dmap = DimensionMap()
dmap.set_crs('what')
with self.assertRaises(DimensionMapError):
dmap.get_group(['level_1'])
level_1 = DimensionMap()
level_1.set_variable(DMK.LEVEL, 'level_1')
dmap.set_group('level_1', level_1)
desired = {'crs': {'variable': 'what'},
'groups': {'level_1': {'level': {'attrs': {},
'bounds': None,
'dimension': [],
'variable': 'level_1'}}}}
actual = dmap.as_dict()
self.assertEqual(actual, desired)
level_1_1 = DimensionMap()
level_1_1.set_variable(DMK.X, 'longitude')
level_1.set_group('level_1_1', level_1_1)
desired = {'crs': {'variable': 'what'},
'groups': {'level_1': {'groups': {'level_1_1': {'x': {'attrs': {},
'bounds': None,
'dimension': [],
'variable': 'longitude'}}},
'level': {'attrs': {},
'bounds': None,
'dimension': [],
'variable': 'level_1'}}}}
actual = dmap.as_dict()
self.assertEqual(actual, desired)
def test_get_group(self):
dmap = DimensionMap()
dmap.set_crs('who')
actual = dmap.get_group(None)
self.assertEqual(dmap.as_dict(), actual.as_dict())
dmap = DimensionMap()
dmap.set_crs('what')
with self.assertRaises(DimensionMapError):
dmap.get_group(['level_1'])
level_1 = DimensionMap()
level_1.set_variable(DMK.LEVEL, 'level_1')
dmap.set_group('level_1', level_1)
desired = {
'crs': {
'variable': 'what'
},
'groups': {
'level_1': {
'level': {
'attrs': {},
'bounds': None,
'dimension': [],
'variable': 'level_1'
}
}
}
}
actual = dmap.as_dict()
self.assertEqual(actual, desired)
level_1_1 = DimensionMap()
level_1_1.set_variable(DMK.X, 'longitude')
level_1.set_group('level_1_1', level_1_1)
desired = {
'crs': {
'variable': 'what'
},
'groups': {
'level_1': {
'groups': {
'level_1_1': {
'x': {
'attrs': {},
'bounds': None,
'dimension': [],
'variable': 'longitude'
}
}
},
'level': {
'attrs': {},
'bounds': None,
'dimension': [],
'variable': 'level_1'
}
}
}
}
actual = dmap.as_dict()
self.assertEqual(actual, desired)