def test_iter(self):
program = AlazarProgram()
program._masks = [4, 5, 6]
program.operations = [1, 2, 3]
program.total_length = 13
program.masks = mock.MagicMock(return_value=342)
mask_maker = mock.MagicMock()
a, b, c = program.iter(mask_maker)
self.assertEqual(a, 342)
self.assertEqual(b, [1, 2, 3])
self.assertEqual(c, 13)
program.masks.assert_called_once_with(mask_maker)
def test_masks(self):
program = AlazarProgram()
for name, data in self.masks.items():
program.set_measurement_mask(name, self.sample_factor, *data)
names = []
def make_mask(name, *data):
np.testing.assert_equal(data, self.expected[name])
assert name not in names
names.append(name)
return name
result = program.masks(make_mask)
self.assertEqual(names, result)