From be7134602144889c85329a8c89d5833f681fb819 Mon Sep 17 00:00:00 2001 From: Sam Gardner Date: Wed, 9 Aug 2023 00:06:19 +0000 Subject: [PATCH] codestyle --- tests/calc/test_indices.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tests/calc/test_indices.py b/tests/calc/test_indices.py index 4fa5835496f..77623cbc3ae 100644 --- a/tests/calc/test_indices.py +++ b/tests/calc/test_indices.py @@ -175,13 +175,16 @@ def test_bunkers_motion(): 6.90054376] * units('m/s') assert_almost_equal(motion.flatten(), truth, 8) + def test_bunkers_motion_with_nans(): """Test Bunkers storm motion with observed sounding.""" data = get_upper_air_data(datetime(2016, 5, 22, 0), 'DDC') - data['u_wind'][24:26] = np.nan - data['v_wind'][24:26] = np.nan + u_with_nan = data['u_wind'] + u_with_nan[24:26] = np.nan + v_with_nan = data['v_wind'] + v_with_nan[24:26] = np.nan motion = concatenate(bunkers_storm_motion(data['pressure'], - data['u_wind'], data['v_wind'], + u_with_nan, v_with_nan, data['height'])) truth = [2.09232447, 0.97612357, 11.25513401, 12.85227283, 6.67372924, 6.9141982] * units('m/s')