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)