typo in test_uvbeam

modified: tests/test_uvbeam.py
RadioAstronomySoftwareGroup · Apr 27, 2018 · 90cd1db · 90cd1db
1 parent dd190b4
commit 90cd1db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyuvdata/tests/test_uvbeam.py b/pyuvdata/tests/test_uvbeam.py
@@ -1336,8 +1336,8 @@ def test_get_beam_functions():
     nt.assert_almost_equal(np.sum(power_beam.get_beam_sq_area()), 4. * numfreqs * npix * dOmega)
 
     # Check to make sure only pseudo Stokes I is accepted
-    nt.assert_raises(NotImplementedError, power_beam.get_beam_area, stokes='Q')
-    nt.assert_raises(NotImplementedError, power_beam.get_beam_sq_area, stokes='Q')
+    nt.assert_raises(NotImplementedError, power_beam.get_beam_area, pol='Q')
+    nt.assert_raises(NotImplementedError, power_beam.get_beam_sq_area, pol='Q')
 
     # Check polarization error
     power_beam.polarization_array = [9, 18, 27, -5]