More test fixes

qutech · May 3, 2018 · 69781aa · 69781aa
1 parent 4435a85
commit 69781aa
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/hardware/tabor_dummy_based_tests.py b/tests/hardware/tabor_dummy_based_tests.py
@@ -381,7 +381,7 @@ def test_upload_segment(self):
         # prevent entering and exiting configuration mode
         channel_pair._configuration_guard_count = 2
 
-        segment = self.TaborSegment(np.ones(192+16, dtype=np.uint16), np.zeros(192+16, dtype=np.uint16))
+        segment = self.TaborSegment(np.ones(192+16, dtype=np.uint16), np.zeros(192+16, dtype=np.uint16), None, None)
         segment_binary = segment.get_as_binary()
         with self.assertRaises(ValueError):
             channel_pair._upload_segment(3, segment)
@@ -424,8 +424,8 @@ def test_amend_segments_flush(self):
         channel_pair._segment_hashes = np.array([1, 2, 3, 4], dtype=np.int64)
 
         data = np.ones(192, dtype=np.uint16)
-        segments = [self.TaborSegment(0*data, 1*data),
-                    self.TaborSegment(1*data, 2*data)]
+        segments = [self.TaborSegment(0*data, 1*data, None, None),
+                    self.TaborSegment(1*data, 2*data, None, None)]
 
         channel_pair._amend_segments(segments)
 
@@ -471,8 +471,8 @@ def test_amend_segments_iter(self):
         channel_pair._segment_hashes = np.array([1, 2, 3, 4], dtype=np.int64)
 
         data = np.ones(192, dtype=np.uint16)
-        segments = [self.TaborSegment(0*data, 1*data),
-                    self.TaborSegment(1*data, 2*data)]
+        segments = [self.TaborSegment(0*data, 1*data, None, None),
+                    self.TaborSegment(1*data, 2*data, None, None)]
 
         indices = channel_pair._amend_segments(segments)