From 4189805240d04ee13996c75264ed087d8209f311 Mon Sep 17 00:00:00 2001 From: Josh Dillon Date: Mon, 24 Feb 2025 11:50:28 -0800 Subject: [PATCH] add test for waterfall_blacklist --- hera_cal/tests/test_smooth_cal.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/hera_cal/tests/test_smooth_cal.py b/hera_cal/tests/test_smooth_cal.py index ab497390b..9ff9a8cc0 100644 --- 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()