add test for waterfall_blacklist

HERA-Team · Feb 24, 2025 · 4189805 · 4189805
1 parent 96ca460
commit 4189805
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/hera_cal/tests/test_smooth_cal.py b/hera_cal/tests/test_smooth_cal.py
@@ -420,6 +420,9 @@ def test_build_wgts_grid(self):
         wgts_grid = smooth_cal._build_wgts_grid(flag_grid, time_blacklist=[False, True], freq_blacklist=[False, False, True], blacklist_wgt=.1)
         np.testing.assert_array_equal(wgts_grid, [[0, 1, .1], [.1, .1, .1]])
 
+        wgts_grid = smooth_cal._build_wgts_grid(flag_grid, waterfall_blacklist=[[False, True, False], [False, False, False]], blacklist_wgt=.1)
+        np.testing.assert_array_equal(wgts_grid, [[0, .1, 1], [1, 1, 1]])
+
     def test_to_anflags(self):
         calfits_list = sorted(glob.glob(os.path.join(DATA_PATH, 'test_input/*.abs.calfits_54x_only')))[0::2]
         uvc = UVCal()