Updating test constructor to deal with change in behavior with warnin…

…gs about extra_keywords in UVBeam
RadioAstronomySoftwareGroup · Jan 31, 2025 · 48acf3a · 48acf3a
1 parent 2b36b9f
commit 48acf3a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_uvsim.py b/tests/test_uvsim.py
@@ -67,6 +67,13 @@ def multi_beams():
         ):
             beam5.to_healpix(nside=8)
         beams.append(beam5)
+    except AssertionError:
+        # If we have an assert error, that means that import worked, but the warning
+        # was not raised due to the change in UVBeam warning behavior. Try again...
+        # TODO: Simplify this once pyuvdata v3.2 is required
+        with check_warnings(None):
+            beam5.to_healpix(nside=8)
+        beams.append(beam5)
     except ImportError:
         pass