Fix tests to find data directory

HERA-Team · Feb 20, 2018 · f4c1f1d · f4c1f1d
1 parent 3588a72
commit f4c1f1d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/hera_pspec/tests/test_pspecdata.py b/hera_pspec/tests/test_pspecdata.py
@@ -230,17 +230,17 @@ def test_get_G(self):
 
     def test_scalar(self):
         dfiles = [
-            'data/zen.2458042.12552.xx.HH.uvXAA',
-            'data/zen.2458042.12552.xx.HH.uvXAA'
+            'zen.2458042.12552.xx.HH.uvXAA',
+            'zen.2458042.12552.xx.HH.uvXAA'
         ]
-        beamfile = 'data/NF_HERA_Beams.beamfits'
+        beamfile = DATADIR + 'NF_HERA_Beams.beamfits'
         d = []
         for dfile in dfiles:
             _d = uv.UVData()
-            _d.read_miriad(dfile)
+            _d.read_miriad(DATADIR + dfile)
             d.append(_d)
         w = [None for _d in dfiles]
-        self.ds = pspecdata.PSpecData(dsets=d, wgts=w,beam=beamfile)
+        self.ds = pspecdata.PSpecData(dsets=d, wgts=w, beam=beamfile)
 
         scalar = self.ds.scalar()
         scalar_double_steps = self.ds.scalar(num_steps=20000) # convergence of integral