def test_same_two(self): source_var = netcdfVarMin(('a', 'b'), (3, 5)) target_var = netcdfVarMin(('a', 'b'), (3, 5)) expected = OrderedDict((('a', 3), ('b', 5))) result = bald.netcdf_shared_dimensions(source_var, target_var) with self.subTest(): self.assertDictEqual(result.get('sourceReshape', dict()), expected) with self.subTest(): self.assertDictEqual(result.get('targetReshape', dict()), expected)
def test_disjoint_misalign_invert(self): source_var = netcdfVarMin(('b', 'c', 'd'), (5, 7, 9)) target_var = netcdfVarMin(('a', 'b', 'c'), (3, 5, 7)) expected_source = OrderedDict((('a', 1), ('b', 5), ('c', 7), ('d', 9))) expected_target = OrderedDict((('a', 3), ('b', 5), ('c', 7), ('d', 1))) result = bald.netcdf_shared_dimensions(source_var, target_var) with self.subTest(): self.assertDictEqual(result.get('sourceReshape', dict()), expected_source) with self.subTest(): self.assertDictEqual(result.get('targetReshape', dict()), expected_target)