Merge bb17f10 into 4ff1a3e

pyuvdata/tests/test_uvbeam.py

@@ -178,6 +178,44 @@ def test_peak_normalize():
     nt.assert_raises(NotImplementedError, power_beam.peak_normalize)
 
 
+def test_stokes_matrix():
+    beam = UVBeam()
+    nt.assert_raises(ValueError, beam._stokes_matrix, -2)
+    nt.assert_raises(ValueError, beam._stokes_matrix, 5)
+
+
+def test_efield_to_pstokes():
+    efield_beam = UVBeam()
+    efield_beam.read_cst_beam(cst_files, beam_type='efield', frequency=[150e6, 123e6],
+                              telescope_name='TEST', feed_name='bob',
+                              feed_version='0.1', feed_pol=['x'],
+                              model_name='E-field pattern - Rigging height 4.9m',
+                              model_version='1.0')
+
+    pstokes_beam = copy.deepcopy(efield_beam)
+    pstokes_beam.az_za_to_healpix()
+    pstokes_beam.efield_to_pstokes()
+
+    pstokes_beam = copy.deepcopy(efield_beam)
+    pstokes_beam.az_za_to_healpix()
+    beam_return = pstokes_beam.efield_to_pstokes(inplace=False)
+
+    pstokes_beam = copy.deepcopy(efield_beam)
+    nt.assert_raises(ValueError, pstokes_beam.efield_to_pstokes)
+
+    pstokes_beam = copy.deepcopy(efield_beam)
+
+    power_beam = UVBeam()
+    power_beam.read_cst_beam(cst_files, beam_type='power', frequency=[150e6, 123e6],
+                             telescope_name='TEST', feed_name='bob',
+                             feed_version='0.1', feed_pol=['x'],
+                             model_name='E-field pattern - Rigging height 4.9m',
+                             model_version='1.0')
+    nt.assert_raises(ValueError, power_beam.efield_to_pstokes)
+
+    nt.assert_raises(ValueError, power_beam.efield_to_pstokes)
+
+
 def test_efield_to_power():
     efield_beam = UVBeam()
     efield_beam.read_cst_beam(cst_files, beam_type='efield', frequency=[150e6, 123e6],
@@ -1373,10 +1411,21 @@ def test_get_beam_functions():
                              model_name='E-field pattern - Rigging height 4.9m',
                              model_version='1.0')
 
-    nt.assert_raises(AssertionError, power_beam._get_beam, 'I')
+    nt.assert_raises(AssertionError, power_beam._get_beam, 'xx')
 
     # Check only healpix accepted (HEALPix checks are in test_healpix)
     # change data_normalization to peak for rest of checks
     power_beam.peak_normalize()
     nt.assert_raises(ValueError, power_beam.get_beam_area)
     nt.assert_raises(ValueError, power_beam.get_beam_sq_area)
+
+    power_beam = UVBeam()
+    power_beam.read_cst_beam(cst_files[0], beam_type='power', frequency=150e6,
+                             telescope_name='TEST', feed_name='bob',
+                             feed_version='0.1',
+                             model_name='E-field pattern - Rigging height 4.9m',
+                             model_version='1.0')
+    power_beam.az_za_to_healpix()
+    power_beam.peak_normalize()
+    power_beam._get_beam('xx')
+    nt.assert_raises(ValueError, power_beam._get_beam, 4)