def test_arrange_along_xyt(self, create_filepaths):
paths = create_filepaths(nxpe=3, nype=2, nt=2)
expected_path_grid = [
[
[
"./run0/BOUT.dmp.0.nc",
"./run0/BOUT.dmp.1.nc",
"./run0/BOUT.dmp.2.nc",
],
[
"./run0/BOUT.dmp.3.nc",
"./run0/BOUT.dmp.4.nc",
"./run0/BOUT.dmp.5.nc",
],
],
[
[
"./run1/BOUT.dmp.0.nc",
"./run1/BOUT.dmp.1.nc",
"./run1/BOUT.dmp.2.nc",
],
[
"./run1/BOUT.dmp.3.nc",
"./run1/BOUT.dmp.4.nc",
"./run1/BOUT.dmp.5.nc",
],
],
]
actual_path_grid, actual_concat_dims = _arrange_for_concatenation(
paths, nxpe=3, nype=2)
assert expected_path_grid == actual_path_grid
assert actual_concat_dims == ["t", "y", "x"]
def test_arrange_single(self, create_filepaths):
paths = create_filepaths(nxpe=1, nype=1, nt=1)
expected_path_grid = [[["./run0/BOUT.dmp.0.nc"]]]
actual_path_grid, actual_concat_dims = _arrange_for_concatenation(
paths, nxpe=1, nype=1)
assert expected_path_grid == actual_path_grid
assert actual_concat_dims == [None, None, None]
def test_arrange_along_xt(self, create_filepaths):
paths = create_filepaths(nxpe=3, nype=1, nt=2)
expected_path_grid = [[['./run0/BOUT.dmp.0.nc', './run0/BOUT.dmp.1.nc', './run0/BOUT.dmp.2.nc']],
[['./run1/BOUT.dmp.0.nc', './run1/BOUT.dmp.1.nc', './run1/BOUT.dmp.2.nc']]]
actual_path_grid, actual_concat_dims = _arrange_for_concatenation(
paths, nxpe=3, nype=1)
assert expected_path_grid == actual_path_grid
assert actual_concat_dims == ['t', None, 'x']
def test_arrange_along_t(self, create_filepaths):
paths = create_filepaths(nxpe=1, nype=1, nt=3)
expected_path_grid = [
[["./run0/BOUT.dmp.0.nc"]],
[["./run1/BOUT.dmp.0.nc"]],
[["./run2/BOUT.dmp.0.nc"]],
]
actual_path_grid, actual_concat_dims = _arrange_for_concatenation(
paths, nxpe=1, nype=1)
assert expected_path_grid == actual_path_grid
assert actual_concat_dims == ["t", None, None]
def test_arrange_along_y(self, create_filepaths):
paths = create_filepaths(nxpe=1, nype=3, nt=1)
expected_path_grid = [[
["./run0/BOUT.dmp.0.nc"],
["./run0/BOUT.dmp.1.nc"],
["./run0/BOUT.dmp.2.nc"],
]]
actual_path_grid, actual_concat_dims = _arrange_for_concatenation(
paths, nxpe=1, nype=3)
assert expected_path_grid == actual_path_grid
assert actual_concat_dims == [None, "y", None]
