def test_concat_once(self, create_combined_ids, concat_dim): shape = (2, ) combined_ids = create_combined_ids(shape) ds = create_test_data result = _combine_all_along_first_dim(combined_ids, dim=concat_dim, data_vars='all', coords='different', compat='no_conflicts') expected_ds = concat([ds(0), ds(1)], dim=concat_dim) assert_combined_tile_ids_equal(result, {(): expected_ds})
def test_concat_only_first_dim(self, create_combined_ids): shape = (2, 3) combined_ids = create_combined_ids(shape) result = _combine_all_along_first_dim(combined_ids, dim='dim1', data_vars='all', coords='different', compat='no_conflicts') ds = create_test_data partway1 = concat([ds(0), ds(3)], dim='dim1') partway2 = concat([ds(1), ds(4)], dim='dim1') partway3 = concat([ds(2), ds(5)], dim='dim1') expected_datasets = [partway1, partway2, partway3] expected = {(i, ): ds for i, ds in enumerate(expected_datasets)} assert_combined_tile_ids_equal(result, expected)