def test_infer_boundaries_2d_parallelization_doublenull(
self, xproc, yproc, nxpe, nype, lower_boundaries,
upper_boundaries):
"""
Numbering scheme for nxpe=3, nype=4
y 9 10 11
^ 6 7 8
| 3 4 5
| 0 1 2
-----> x
"""
ds = create_test_data(0)
ds["jyseps2_1"] = 3
ds["jyseps1_2"] = 11
ds["ny_inner"] = 8
ds["MYSUB"] = 4
ds["PE_XIND"] = xproc
ds["PE_YIND"] = yproc
actual_lower_boundaries, actual_upper_boundaries = _infer_contains_boundaries(
ds, nxpe, nype)
assert actual_lower_boundaries == lower_boundaries
assert actual_upper_boundaries == upper_boundaries
def test_infer_boundaries_2d_parallelization_doublenull_by_filenum(
self, xproc, yproc, nxpe, nype, lower_boundaries,
upper_boundaries):
"""
Numbering scheme for nxpe=3, nype=4
y 9 10 11
^ 6 7 8
| 3 4 5
| 0 1 2
-----> x
"""
filenum = yproc * nxpe + xproc
ds = create_test_data(0)
ds["jyseps2_1"] = 3
ds["jyseps1_2"] = 11
ds["ny_inner"] = 8
ds["MYSUB"] = 4
ds.encoding["source"] = "folder0/BOUT.dmp." + str(filenum) + ".nc"
actual_lower_boundaries, actual_upper_boundaries = _infer_contains_boundaries(
ds, nxpe, nype)
assert actual_lower_boundaries == lower_boundaries
assert actual_upper_boundaries == upper_boundaries
def test_infer_boundaries_2d_parallelization(
self, xproc, yproc, nxpe, nype, lower_boundaries, upper_boundaries):
"""
Numbering scheme for nxpe=3, nype=4
y 9 10 11
^ 6 7 8
| 3 4 5
| 0 1 2
-----> x
"""
ds = create_test_data(0)
ds['jyseps2_1'] = 0
ds['jyseps1_2'] = 0
ds['PE_XIND'] = xproc
ds['PE_YIND'] = yproc
actual_lower_boundaries, actual_upper_boundaries = _infer_contains_boundaries(
ds, nxpe, nype)
assert actual_lower_boundaries == lower_boundaries
assert actual_upper_boundaries == upper_boundaries
