FIX: nsweeps wasn't updated when calling 'determine_sweeps'; tests we…

…re updated accordingly
ARM-DOE · Dec 6, 2023 · 1d338c1 · 1d338c1
1 parent ea1a94b
commit 1d338c1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/pyart/util/radar_utils.py b/pyart/util/radar_utils.py
@@ -182,6 +182,7 @@ def determine_sweeps(radar,
     radar.sweep_start_ray_index['data'] = ma.array(sweep_start_index, dtype='int32')
     radar.sweep_end_ray_index['data'] = ma.array(sweep_end_index, dtype='int32')
     radar.sweep_number['data'] = ma.array(sweep_number, dtype='int32')
+    radar.nsweeps = len(sweep_number)
     return
 
 

diff --git a/tests/util/test_radar_utils.py b/tests/util/test_radar_utils.py
@@ -58,6 +58,7 @@ def test_determine_sweeps():
     assert np.all(radar.sweep_end_ray_index['data'] == [35, 71, 107])
     assert np.all(radar.sweep_start_ray_index['data'] == [0, 36, 72])
     assert len(radar.sweep_number['data']) == 3
+    assert radar.nsweeps == 3
 
     # rhi
     radar = pyart.testing.make_empty_rhi_radar(10, 25, 5)
@@ -66,6 +67,7 @@ def test_determine_sweeps():
     assert np.all(radar.sweep_end_ray_index['data'] == [ 24,  49,  74,  99, 124])
     assert np.all(radar.sweep_start_ray_index['data'] == [  0,  25,  50,  75, 100])
     assert len(radar.sweep_number['data']) == 5
+    assert radar.nsweeps == 5
 
 
 def test_subset_radar():