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)
