diff --git a/src/cr/cube/cube_slice.py b/src/cr/cube/cube_slice.py index 8751949da..09e9d2c8b 100644 --- a/src/cr/cube/cube_slice.py +++ b/src/cr/cube/cube_slice.py @@ -262,8 +262,9 @@ def scale_means(self, hs_dims=None, prune=False): scaled mean (as numpy array). If both row and col scaled means are present, return them as two numpy arrays inside of a list. """ + scale_means = self._cube.scale_means(hs_dims, prune) if self.ca_as_0th: - return [None, None] + return [scale_means[0][-1][self._index]] return self._cube.scale_means(hs_dims, prune)[self._index] @memoize diff --git a/tests/unit/test_cube_slice.py b/tests/unit/test_cube_slice.py index 38e7e834f..b2112c426 100644 --- a/tests/unit/test_cube_slice.py +++ b/tests/unit/test_cube_slice.py @@ -574,8 +574,10 @@ def test_scale_means_for_ca_as_0th(self): """Test that CA as 0th slice always returns empty scale means.""" cube = Mock() cube.dim_types = (DT.CA_SUBVAR,) + scale_means_value = Mock() + cube.scale_means.return_value = [[None, [scale_means_value, Mock(), Mock()]]] cs = CubeSlice(cube, 0, ca_as_0th=True) - assert cs.scale_means() == [None, None] + assert cs.scale_means() == [scale_means_value] def test_shape_property_deprecated(self): cube = Mock()