Only fixing turbulence function

blue-jaye-121 · blue-jaye-121 · commit e518adc38d14 · 2025-07-24T15:15:38.000-06:00
diff --git a/src/metpy/calc/basic.py b/src/metpy/calc/basic.py
@@ -631,7 +631,7 @@ def coriolis_parameter(latitude):
 
 
 @exporter.export
-@preprocess_and_wrap()
+@preprocess_and_wrap(wrap_like='pressure')
 @check_units('[pressure]', '[length]')
 def add_height_to_pressure(pressure, height):
     r"""Calculate the pressure at a certain height above another pressure level.
@@ -1224,7 +1224,7 @@ def altimeter_to_station_pressure(altimeter_value, height):
 
 
 @exporter.export
-@preprocess_and_wrap()
+@preprocess_and_wrap(wrap_like='altimeter_value')
 @check_units('[pressure]', '[length]', '[temperature]')
 def altimeter_to_sea_level_pressure(altimeter_value, height, temperature):
     r"""Convert the altimeter setting to sea-level pressure.
diff --git a/src/metpy/calc/thermo.py b/src/metpy/calc/thermo.py
@@ -517,7 +517,10 @@ def dry_lapse(pressure, temperature, reference_pressure=None, vertical_dim=0):
 
 
 @exporter.export
-@preprocess_and_wrap()
+@preprocess_and_wrap(
+    wrap_like='temperature',
+    broadcast=('pressure', 'temperature', 'reference_pressure')
+)
 @process_units(
     {
         'pressure': '[pressure]',
diff --git a/tests/calc/test_turbulence.py b/tests/calc/test_turbulence.py
@@ -31,6 +31,7 @@ def uvw_and_known_tke():
     e_true = 30
     return u, v, w, e_true
 
+
 @pytest.fixture()
 def uvw_and_known_tke_xarray():
     """Provide a set of u, v, w with a known tke value as an xarray"""
@@ -39,20 +40,20 @@ def uvw_and_known_tke_xarray():
     lat = [10, 20]          # degrees North
     lon = [30, 40]          # degrees East
     time = np.array(['2025-01-01T00:00', '2025-01-01T06:00'], dtype='datetime64')
-    
+
     # Define dimensions
     dims = ("pressure", "lat", "lon", "time")
-    
+
     # Generate 16 linearly spaced values between -30 and 30
     uwind_values = np.linspace(0, 30, num=16).reshape(2, 2, 2, 2)
     vwind_values = np.linspace(-30, 0, num=16).reshape(2, 2, 2, 2)
     wwind_values = np.linspace(-1, 2, num=16).reshape(2, 2, 2, 2)
-    
+
     # Apply units
     u = uwind_values * units('m/s')
     v = vwind_values * units('m/s')
     w = wwind_values * units('m/s')
-    
+
     # Create the Dataset
     ds = xr.Dataset(
         {
@@ -66,11 +67,12 @@ def uvw_and_known_tke_xarray():
             "lon": lon,
             "time": time
         }
-        )   
-     
-    e_true = np.full((2, 2, 2), 1.005)  * units("m^2/s^2")
+    )
+
+    e_true = np.full((2, 2, 2), 1.005) * units("m^2/s^2")
     return ds, e_true
 
+
 def test_no_tke_1d():
     """Test tke calculation where the expected value is 0."""
     observations = 5
@@ -110,8 +112,8 @@ def test_known_tke(uvw_and_known_tke):
     """Test basic behavior of tke with known values."""
     u, v, w, e_true = uvw_and_known_tke
     assert_array_equal(e_true, tke(u, v, w))
-    
-    
+
+
 def test_known_tke_xarray(uvw_and_known_tke_xarray):
     """Test basic behavior of tke with known xarray values"""
     data, e_true = uvw_and_known_tke_xarray
@@ -380,6 +382,7 @@ def uvw_and_known_u_star_zero_mean():
     u_star_true = {'uw': 2.0, 'uwvw': 2.3784142300054421}
     return u, v, w, u_star_true
 
+
 @pytest.fixture()
 def uvw_and_known_friction_velocity_xarray():
     """Provide a set of u, v, w with a known tke value as an xarray"""
@@ -388,20 +391,20 @@ def uvw_and_known_friction_velocity_xarray():
     lat = [10, 20]          # degrees North
     lon = [30, 40]          # degrees East
     time = np.array(['2025-01-01T00:00', '2025-01-01T06:00'], dtype='datetime64')
-    
+
     # Define dimensions
     dims = ("pressure", "lat", "lon", "time")
-    
+
     # Generate 16 linearly spaced values between -30 and 30
     uwind_values = np.linspace(0, 30, num=16).reshape(2, 2, 2, 2)
     vwind_values = np.linspace(-30, 0, num=16).reshape(2, 2, 2, 2)
     wwind_values = np.linspace(-1, 2, num=16).reshape(2, 2, 2, 2)
-    
+
     # Apply units
     u = uwind_values * units('m/s')
     v = vwind_values * units('m/s')
     w = wwind_values * units('m/s')
-    
+
     # Create the Dataset
     ds = xr.Dataset(
         {
@@ -415,10 +418,11 @@ def uvw_and_known_friction_velocity_xarray():
             "lon": lon,
             "time": time
         }
-    )   
+    )
     expected = np.full((2, 2, 2), .3760603) * units('meter / second')
     return ds, expected
 
+
 @pytest.fixture()
 def uvw_and_known_u_star_nonzero_mean():
     """Return components and friction velocity for a non-zero-mean time series."""
@@ -445,10 +449,11 @@ def test_u_star_1d_nonzero_mean(uvw_and_known_u_star_nonzero_mean):
                         u_star_true['uw'])
     assert_almost_equal(friction_velocity(u, w, v=v, perturbation=False),
                         u_star_true['uwvw'])
-    
+
+
 def test_friction_velocity_nonzero_xarray(uvw_and_known_friction_velocity_xarray):
     """Test friction velocity in 1d with an xarray"""
-    data, expected = uvw_and_known_friction_velocity_xarray;
+    data, expected = uvw_and_known_friction_velocity_xarray
     assert_array_almost_equal(friction_velocity(data.uwind, data.wwind, data.vwind), expected)
 
 
