Missing tests for coverage

Author: adeliegorce

hera_pspec/tests/test_pspecbeam.py

@@ -4,6 +4,7 @@
 import pyuvdata as uv
 from .. import pspecbeam, conversions
 from hera_pspec.data import DATA_PATH
+from pyuvdata import UVBeam
 
 
 class Test_DataSet(unittest.TestCase):
@@ -25,7 +26,9 @@ def test_UVbeam(self):
         # Precomputed results in the following tests were done "by hand" using
         # iPython notebook "Scalar_dev2.ipynb" in tests directory
         pstokes_beamfile = os.path.join(DATA_PATH, "HERA_NF_pstokes_power.beamfits")
-        beam = pspecbeam.PSpecBeamUV(pstokes_beamfile)
+        uvb = UVBeam()
+        uvb.read_beamfits(pstokes_beamfile, use_future_array_shapes=True)
+        beam = pspecbeam.PSpecBeamUV(uvb)
         Om_p = beam.power_beam_int()
         Om_pp = beam.power_beam_sq_int()
         lower_freq = 120.*10**6

hera_pspec/tests/test_testing.py

@@ -130,15 +130,15 @@ def test_sky_noise_jy_autos():
 
 
 def test_sky_noise_sim():
-    uvd = UVData()
     uvfile = os.path.join(DATA_PATH, "zen.even.xx.LST.1.28828.uvOCRSA")
+    uvd = UVData()
     uvd.read_miriad(uvfile, use_future_array_shapes=True)
     beam = os.path.join(DATA_PATH, "HERA_NF_dipole_power.beamfits")
     beam_ps = os.path.join(DATA_PATH, "HERA_NF_pstokes_power.beamfits")
 
     # basic test
     np.random.seed(0)
-    sim = testing.sky_noise_sim(uvd, beam, cov_amp=1000, cov_length_scale=10, constant_in_time=True,
+    sim = testing.sky_noise_sim(uvfile, beam, cov_amp=1000, cov_length_scale=10, constant_in_time=True,
                                 divide_by_nsamp=False)
     # assert something was inserted
     for bl in sim.get_antpairpols():