def setUp(self):
# set up a pulsar with some mock settings for the element
self.pulsar = Pulsar()
for i in range(4):
self.pulsar.define_channel(
id='ch{}'.format(i + 1),
name='ch{}'.format(i + 1),
type='analog',
# max safe IQ voltage
high=.7,
low=-.7,
offset=0.0,
delay=0,
active=True)
self.pulsar.define_channel(id='ch{}_marker1'.format(i + 1),
name='ch{}_marker1'.format(i + 1),
type='marker',
high=2.0,
low=0,
offset=0.,
delay=0,
active=True)
self.pulsar.define_channel(id='ch{}_marker2'.format(i + 1),
name='ch{}_marker2'.format(i + 1),
type='marker',
high=2.0,
low=0,
offset=0.,
delay=0,
active=True)
# We need to discuss where to store this stuff
fsqs.kernel_dir = 'testing/kernels/'
self.kernel_list = ['kernels_rabi/kernel_160208_220046_it1.txt']
def setUp(self):
# set up a pulsar with some mock settings for the element
self.pulsar = Pulsar()
for i in range(4):
self.pulsar.define_channel(
id="ch{}".format(i + 1),
name="ch{}".format(i + 1),
type="analog",
# max safe IQ voltage
high=0.7,
low=-0.7,
offset=0.0,
delay=0,
active=True,
)
self.pulsar.define_channel(
id="ch{}_marker1".format(i + 1),
name="ch{}_marker1".format(i + 1),
type="marker",
high=2.0,
low=0,
offset=0.0,
delay=0,
active=True,
)
self.pulsar.define_channel(
id="ch{}_marker2".format(i + 1),
name="ch{}_marker2".format(i + 1),
type="marker",
high=2.0,
low=0,
offset=0.0,
delay=0,
active=True,
)
# We need to discuss where to store this stuff
fsqs.kernel_dir_path = "testing/kernels/"
self.kernel_list = ["kernels_rabi/kernel_160208_220046_it1.txt"]
class element_distortion(unittest.TestCase):
def setUp(self):
# set up a pulsar with some mock settings for the element
self.pulsar = Pulsar()
for i in range(4):
self.pulsar.define_channel(
id='ch{}'.format(i + 1),
name='ch{}'.format(i + 1),
type='analog',
# max safe IQ voltage
high=.7,
low=-.7,
offset=0.0,
delay=0,
active=True)
self.pulsar.define_channel(id='ch{}_marker1'.format(i + 1),
name='ch{}_marker1'.format(i + 1),
type='marker',
high=2.0,
low=0,
offset=0.,
delay=0,
active=True)
self.pulsar.define_channel(id='ch{}_marker2'.format(i + 1),
name='ch{}_marker2'.format(i + 1),
type='marker',
high=2.0,
low=0,
offset=0.,
delay=0,
active=True)
# We need to discuss where to store this stuff
fsqs.kernel_dir = 'testing/kernels/'
self.kernel_list = ['kernels_rabi/kernel_160208_220046_it1.txt']
def test_distorted_attribute(self):
test_elt = element.Element('test_elt', pulsar=self.pulsar)
self.assertTrue((len(test_elt._channels)) != 0)
for ch, item in test_elt._channels.items():
self.assertFalse(item['distorted'])
self.assertEqual(len(test_elt.distorted_wfs), 0)
test_elt.add(
SquarePulse(name='dummy_sqaure',
channel='ch1',
amplitude=.3,
length=20e-9))
dist_dict = {'ch_list': ['ch1'], 'ch1': self.kernel_list}
test_elt = fsqs.distort(test_elt, dist_dict)
self.assertEqual(len(test_elt.distorted_wfs), 1)
for ch, item in test_elt._channels.items():
if ch == 'ch1':
self.assertTrue(item['distorted'])
self.assertTrue(ch in test_elt.distorted_wfs.keys())
else:
self.assertFalse(item['distorted'])
class element_distortion(unittest.TestCase):
def setUp(self):
# set up a pulsar with some mock settings for the element
self.pulsar = Pulsar()
for i in range(4):
self.pulsar.define_channel(
id="ch{}".format(i + 1),
name="ch{}".format(i + 1),
type="analog",
# max safe IQ voltage
high=0.7,
low=-0.7,
offset=0.0,
delay=0,
active=True,
)
self.pulsar.define_channel(
id="ch{}_marker1".format(i + 1),
name="ch{}_marker1".format(i + 1),
type="marker",
high=2.0,
low=0,
offset=0.0,
delay=0,
active=True,
)
self.pulsar.define_channel(
id="ch{}_marker2".format(i + 1),
name="ch{}_marker2".format(i + 1),
type="marker",
high=2.0,
low=0,
offset=0.0,
delay=0,
active=True,
)
# We need to discuss where to store this stuff
fsqs.kernel_dir_path = "testing/kernels/"
self.kernel_list = ["kernels_rabi/kernel_160208_220046_it1.txt"]
def test_distorted_attribute(self):
test_elt = element.Element("test_elt", pulsar=self.pulsar)
self.assertTrue((len(test_elt._channels)) != 0)
for ch, item in test_elt._channels.items():
self.assertFalse(item["distorted"])
self.assertEqual(len(test_elt.distorted_wfs), 0)
test_elt.add(SquarePulse(name="dummy_sqaure", channel="ch1", amplitude=0.3, length=20e-9))
dist_dict = {"ch_list": ["ch1"], "ch1": self.kernel_list}
test_elt = fsqs.distort(test_elt, dist_dict)
self.assertEqual(len(test_elt.distorted_wfs), 1)
for ch, item in test_elt._channels.items():
if ch == "ch1":
self.assertTrue(item["distorted"])
self.assertTrue(ch in test_elt.distorted_wfs.keys())
else:
self.assertFalse(item["distorted"])
