change to test to accomodate prev commit

RadioAstronomySoftwareGroup · Jun 1, 2020 · 1b51565 · 1b51565
1 parent 087c9b0
commit 1b51565
Showing 1 changed file with 2 additions and 5 deletions.
diff --git 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):