Skip to content

Commit

Permalink
Rename manipulate_realization_dimension plugin to manipulate_n_realiz…
Browse files Browse the repository at this point in the history
…ations.
  • Loading branch information
brhooper committed Oct 10, 2024
1 parent aa24f24 commit b37cf83
Show file tree
Hide file tree
Showing 4 changed files with 9 additions and 9 deletions.
4 changes: 2 additions & 2 deletions improver/cli/manipulate_n_realizations.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,8 @@ def process(
iris.cube.Cube:
The processed cube.
"""
from improver.utilities.cube_manipulation import manipulate_realization_dimension
from improver.utilities.cube_manipulation import manipulate_n_realizations

output = manipulate_realization_dimension(cube, n_realizations)
output = manipulate_n_realizations(cube, n_realizations)

return output
4 changes: 2 additions & 2 deletions improver/ensemble_copula_coupling/ensemble_copula_coupling.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@
from improver.utilities.cube_manipulation import (
enforce_coordinate_ordering,
get_dim_coord_names,
manipulate_realization_dimension,
manipulate_n_realizations,
)
from improver.utilities.indexing_operations import choose

Expand Down Expand Up @@ -1338,7 +1338,7 @@ def _recycle_raw_ensemble_realizations(
if plen == mlen:
pass
else:
raw_forecast_realizations = manipulate_realization_dimension(
raw_forecast_realizations = manipulate_n_realizations(
raw_forecast_realizations, plen
)
return raw_forecast_realizations
Expand Down
2 changes: 1 addition & 1 deletion improver/utilities/cube_manipulation.py
Original file line number Diff line number Diff line change
Expand Up @@ -826,7 +826,7 @@ def height_of_maximum(
return height_of_max


def manipulate_realization_dimension(cube: Cube, n_realizations: int) -> Cube:
def manipulate_n_realizations(cube: Cube, n_realizations: int) -> Cube:
"""Extend or reduce the number of realizations in a cube.
If more realizations are requested than are in the input cube, then the ensemble
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
import pytest

from improver.synthetic_data.set_up_test_cubes import set_up_variable_cube
from improver.utilities.cube_manipulation import manipulate_realization_dimension
from improver.utilities.cube_manipulation import manipulate_n_realizations


@pytest.fixture
Expand All @@ -26,7 +26,7 @@ def test_basic(temperature_cube, n_realizations):
"""Test that a cube is returned with expected data and realization coordinate."""
input_len = len(temperature_cube.coord("realization").points)
expected_realizations = np.array([r for r in range(n_realizations)])
result = manipulate_realization_dimension(temperature_cube, n_realizations)
result = manipulate_n_realizations(temperature_cube, n_realizations)

assert len(result.coord("realization").points) == n_realizations
assert np.all(result.coord("realization").points == expected_realizations)
Expand All @@ -50,7 +50,7 @@ def test_realizations_start_from_one(temperature_cube):
expected_realizations = [1, 2, 3, 4, 5, 6]
expected_recycling = [1, 2, 3, 1, 2, 3]

result = manipulate_realization_dimension(input_cube, n_realizations)
result = manipulate_n_realizations(input_cube, n_realizations)

assert len(result.coord("realization").points) == n_realizations
assert np.all(result.coord("realization").points == expected_realizations)
Expand All @@ -75,4 +75,4 @@ def test_non_realization_cube(temperature_cube):
)

with pytest.raises(ValueError, match=msg):
manipulate_realization_dimension(temperature_cube, n_realizations=3)
manipulate_n_realizations(temperature_cube, n_realizations=3)

0 comments on commit b37cf83

Please sign in to comment.