From 3e4c35a6b069218d209f8204b6156e674c5bf70e Mon Sep 17 00:00:00 2001
From: Erin Sheldon <erin.sheldon@gmail.com>
Date: Wed, 19 Jul 2023 11:01:50 -0400
Subject: [PATCH] update flags name for test

---
 shear_meas_tests/test_shear_meas.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/shear_meas_tests/test_shear_meas.py b/shear_meas_tests/test_shear_meas.py
index c809ff08..73647cd5 100644
--- a/shear_meas_tests/test_shear_meas.py
+++ b/shear_meas_tests/test_shear_meas.py
@@ -53,8 +53,9 @@ def _make_lsst_sim(*, rng, g1, g2, layout):
 
 
 def _shear_cuts(arr):
+    assert arr is not None
     msk = (
-        (arr['flags'] == 0)
+        (arr['wmom_flags'] == 0)
         & (arr['wmom_s2n'] > 10)
         & (arr['wmom_T_ratio'] > 1.2)
     )