All these `from_global_dim_data_<distribution>` tests are indexing with for loops like this. Should this be changed for the all?