more unit tests

RadioAstronomySoftwareGroup · Jun 27, 2018 · bb17f10 · bb17f10
1 parent 0ae5d16
commit bb17f10
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/pyuvdata/tests/test_uvbeam.py b/pyuvdata/tests/test_uvbeam.py
@@ -196,9 +196,15 @@ def test_efield_to_pstokes():
     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',
@@ -207,6 +213,8 @@ def test_efield_to_pstokes():
                              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()
@@ -1410,4 +1418,14 @@ def test_get_beam_functions():
     power_beam.peak_normalize()
     nt.assert_raises(ValueError, power_beam.get_beam_area)
     nt.assert_raises(ValueError, power_beam.get_beam_sq_area)
-    nt.assert_raises(AssertionError, power_beam._get_beam, 'pI')
+
+    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)