From 1b51565865e2e08c214479e345d584fcbacc64c8 Mon Sep 17 00:00:00 2001 From: Mike Wilensky Date: Mon, 1 Jun 2020 09:47:55 -0700 Subject: [PATCH] change to test to accomodate prev commit --- pyuvdata/uvflag/tests/test_uvflag.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/pyuvdata/uvflag/tests/test_uvflag.py b/pyuvdata/uvflag/tests/test_uvflag.py index 85bb982740..b1670a047a 100644 --- a/pyuvdata/uvflag/tests/test_uvflag.py +++ b/pyuvdata/uvflag/tests/test_uvflag.py @@ -1104,12 +1104,9 @@ def test_to_waterfall_bl_ret_wt_sq(): uvf.to_waterfall(return_weights_square=True) assert np.all(uvf.weights_square_array == 4 * Nbls) - # Check that weights_square_array is required - assert uvf._weights_square_array.required - - # Switch to flag and check that it is now unrequired + # Switch to flag and check that it is now set to None uvf.to_flag() - assert not uvf._weights_square_array.required + assert uvf.weights_square_array is None def test_collapse_pol(test_outfile):