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)