Skip to content

Commit

Permalink
Adapt XPPM translate test to serialization tweaks in Fortran
Browse files Browse the repository at this point in the history
  • Loading branch information
FlorianDeconinck committed Jan 20, 2025
1 parent bd3022d commit f34fc22
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions tests/savepoint/translate/translate_xppm.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,12 +14,12 @@ def __init__(
):
super().__init__(grid, namelist, stencil_factory)
self.in_vars["data_vars"] = {
"q": {"serialname": "qx", "jstart": "jfirst"},
"c": {"serialname": "cx", "istart": grid.is_},
"q": {"serialname": "xppm_q", "jstart": "jfirst"},
"c": {"serialname": "xppm_c", "istart": grid.is_},
}
self.in_vars["parameters"] = ["iord", "jfirst", "jlast"]
self.out_vars = {
"xflux": {
"xppm_flux": {
"istart": grid.is_,
"iend": grid.ie + 1,
"jstart": "jfirst",
Expand All @@ -40,7 +40,7 @@ def process_inputs(self, inputs):

def compute(self, inputs):
self.process_inputs(inputs)
inputs["xflux"] = utils.make_storage_from_shape(
inputs["xppm_flux"] = utils.make_storage_from_shape(
inputs["q"].shape, backend=self.stencil_factory.backend
)
origin = self.grid.grid_indexing.origin_compute()
Expand All @@ -53,7 +53,7 @@ def compute(self, inputs):
origin=(origin[0], int(inputs["jfirst"]), origin[2]),
domain=(domain[0], int(inputs["jlast"] - inputs["jfirst"] + 1), domain[2]),
)
self.compute_func(inputs["q"], inputs["c"], inputs["xflux"])
self.compute_func(inputs["q"], inputs["c"], inputs["xppm_flux"])
return self.slice_output(inputs)


Expand All @@ -65,5 +65,5 @@ def __init__(
stencil_factory: StencilFactory,
):
super().__init__(grid, namelist, stencil_factory)
self.in_vars["data_vars"]["q"]["serialname"] = "q"
self.out_vars["xflux"]["serialname"] = "xflux_2"
self.in_vars["data_vars"]["q"]["serialname"] = "xppm_q2"
self.out_vars["xppm_flux"]["serialname"] = "xppm_flux_2"

0 comments on commit f34fc22

Please sign in to comment.