do not put a Mock in numpys assert_equal

qutech · Apr 2, 2020 · c77770a · c77770a
1 parent d7aaf40
commit c77770a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/hardware/base_tests.py b/tests/hardware/base_tests.py
@@ -80,8 +80,8 @@ def test_init(self):
             sample_waveforms.assert_called_once_with(self.waveforms[:1])
 
     def test_sample_waveforms(self):
-        empty_ch = mock.Mock()
-        empty_m = mock.Mock()
+        empty_ch = np.array([1, 2, 3])
+        empty_m = np.array([0, 1, 0])
         expected_sampled = [
             ((self.sampled[0]['A'], empty_ch, self.sampled[0]['C']), (empty_m, self.sampled[0]['M'] != 0)),
             ((self.sampled[1]['A'], empty_ch, self.sampled[1]['C']), (empty_m, self.sampled[1]['M'] != 0))