def __init__(self, comm=None):
_, rank, _ = get_standard_comm_state(comm=comm)
self.level = None
self.null = True # pass through if not configured
self.parent = None
self.callback = None
self.callback_level = None
self.loggers = None
self.rank = rank
logging.captureWarnings(None)
def test_get_nonempty_ranks(self):
from ocgis.variable.dimension import Dimension
comm, rank, size = get_standard_comm_state()
if size not in [1, 3]:
raise SkipTest('MPI_SIZE != 1 or 3')
target = Dimension('a')
live_ranks = get_nonempty_ranks(target, vm)
if MPI_RANK == 0:
self.assertEqual(live_ranks, tuple(range(size)))
if MPI_SIZE == 3:
targets = {0: Dimension('a', is_empty=True, dist=True),
1: Dimension('a'),
2: Dimension('a')}
with vm.scoped('ner', vm.ranks):
live_ranks = get_nonempty_ranks(targets[MPI_RANK], vm)
self.assertEqual(live_ranks, (1, 2))