added label1 label2 test to pspecdata.pspec output

HERA-Team · Apr 28, 2018 · 482993d · 482993d
1 parent 528f8db
commit 482993d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hera_pspec/tests/test_pspecdata.py b/hera_pspec/tests/test_pspecdata.py
@@ -536,7 +536,7 @@ def test_check_in_dset(self):
     def test_pspec(self):
         # generate ds
         uvd = copy.deepcopy(self.uvd)
-        ds = pspecdata.PSpecData(dsets=[uvd, uvd], wgts=[None, None], beam=self.bm)
+        ds = pspecdata.PSpecData(dsets=[uvd, uvd], wgts=[None, None], beam=self.bm, labels=['red', 'blue'])
 
         # check basic execution with baseline list
         bls = [(24, 25), (37, 38), (38, 39), (52, 53)]