class DataSetInstance(object):
def setUp(self):
super(DataSetInstance, self).setUp()
self.ds = DataSet(outcomeLabels=[('0',), ('1',)])
self.ds.add_count_dict(('Gx', 'Gx'), {('0',): 40, ('1',): 60})
self.ds.add_count_dict(('Gx', 'Gy'), {('0',): 40, ('1',): 60})
self.ds.add_count_dict(('Gy', 'Gx'), {('0',): 40, ('1',): 60})
self.ds.add_count_dict(('Gy', 'Gy'), {('0',): 40, ('1',): 60})
self.ds.done_adding_data()
class StdListTester(BaseCase):
def setUp(self):
self.opLabels = [Label('Gx'), Label('Gy')]
self.strs = cc.circuit_list([('Gx', ), ('Gy', ), ('Gx', 'Gx')])
self.germs = cc.circuit_list([('Gx', 'Gy'), ('Gy', 'Gy')])
self.testFidPairs = [(0, 1)]
self.testFidPairsDict = {
(Label('Gx'), Label('Gy')): [(0, 0), (0, 1)],
(Label('Gy'), Label('Gy')): [(0, 0)]
}
self.ds = DataSet(outcomeLabels=['0',
'1']) # a dataset that is missing
self.ds.add_count_dict(('Gx', ), {
'0': 10,
'1': 90
}) # almost all our strings...
self.ds.done_adding_data()
def test_lsgst_lists_structs(self):
maxLens = [1, 2]
lsgstLists = stdlists.make_lsgst_lists(
std1Q_XY.target_model(),
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="whole germ powers") # also try a Model as first arg
lsgstStructs = stdlists.make_lsgst_structs(
std1Q_XY.target_model(),
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="whole germ powers") # also try a Model as first arg
self.assertEqual(set(lsgstLists[-1]), set(lsgstStructs[-1].allstrs))
lsgstLists2 = stdlists.make_lsgst_lists(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="truncated germ powers")
lsgstStructs2 = stdlists.make_lsgst_structs(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="truncated germ powers")
self.assertEqual(set(lsgstLists2[-1]), set(lsgstStructs2[-1].allstrs))
lsgstLists3 = stdlists.make_lsgst_lists(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="length as exponent")
lsgstStructs3 = stdlists.make_lsgst_structs(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="length as exponent")
self.assertEqual(set(lsgstLists3[-1]), set(lsgstStructs3[-1].allstrs))
maxLens = [1, 2]
lsgstLists4 = stdlists.make_lsgst_lists(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="whole germ powers",
nest=False)
lsgstStructs4 = stdlists.make_lsgst_structs(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="whole germ powers",
nest=False)
self.assertEqual(set(lsgstLists4[-1]), set(lsgstStructs4[-1].allstrs))
lsgstLists5 = stdlists.make_lsgst_lists(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=self.testFidPairs,
truncScheme="whole germ powers")
lsgstStructs5 = stdlists.make_lsgst_structs(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=self.testFidPairs,
truncScheme="whole germ powers")
self.assertEqual(set(lsgstLists5[-1]), set(lsgstStructs5[-1].allstrs))
lsgstLists6 = stdlists.make_lsgst_lists(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=self.testFidPairsDict,
truncScheme="whole germ powers")
lsgstStructs6 = stdlists.make_lsgst_structs(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=self.testFidPairsDict,
truncScheme="whole germ powers")
self.assertEqual(set(lsgstLists6[-1]), set(lsgstStructs6[-1].allstrs))
lsgstLists7 = stdlists.make_lsgst_lists(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="whole germ powers",
keepFraction=0.5,
keepSeed=1234)
lsgstStructs7 = stdlists.make_lsgst_structs(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="whole germ powers",
keepFraction=0.5,
keepSeed=1234)
self.assertEqual(set(lsgstLists7[-1]), set(lsgstStructs7[-1].allstrs))
lsgstLists8 = stdlists.make_lsgst_lists(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=self.testFidPairs,
truncScheme="whole germ powers",
keepFraction=0.7,
keepSeed=1234)
lsgstStructs8 = stdlists.make_lsgst_structs(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=self.testFidPairs,
truncScheme="whole germ powers",
keepFraction=0.7,
keepSeed=1234)
self.assertEqual(set(lsgstLists8[-1]), set(lsgstStructs8[-1].allstrs))
# TODO assert correctness
# empty max-lengths ==> no output
lsgstStructs9 = stdlists.make_lsgst_structs(self.opLabels, self.strs,
self.strs, self.germs, [])
self.assertEqual(len(lsgstStructs9), 0)
# checks against datasets
lgst_strings = cc.list_lgst_circuits(self.strs, self.strs,
self.opLabels)
lsgstStructs10 = stdlists.make_lsgst_structs(self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
dscheck=self.ds,
actionIfMissing="drop",
verbosity=4)
self.assertEqual([Circuit(('Gx', ))], lsgstStructs10[-1].allstrs)
def test_lsgst_experiment_list(self):
maxLens = [1, 2]
lsgstExpList = stdlists.make_lsgst_experiment_list(
self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="whole germ powers")
lsgstExpListb = stdlists.make_lsgst_experiment_list(
std1Q_XY.target_model(),
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="whole germ powers") # with Model as first arg
# TODO assert correctness
def test_lsgst_lists_structs_raises_on_bad_scheme(self):
maxLens = [1, 2]
with self.assertRaises(ValueError):
stdlists.make_lsgst_lists(self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="foobar")
with self.assertRaises(ValueError):
stdlists.make_lsgst_structs(self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
fidPairs=None,
truncScheme="foobar")
with self.assertRaises(ValueError):
stdlists.make_lsgst_structs(self.opLabels,
self.strs,
self.strs,
self.germs,
maxLens,
dscheck=self.ds,
actionIfMissing="foobar")
def test_lsgst_lists_structs_raises_on_missing_ds_sequence(self):
with self.assertRaises(ValueError):
stdlists.make_lsgst_structs(self.opLabels,
self.strs,
self.strs,
self.germs, [1, 2],
dscheck=self.ds) # missing sequences
def test_elgst_lists_structs(self):
# ELGST
maxLens = [1, 2]
elgstLists = stdlists.make_elgst_lists(self.opLabels,
self.germs,
maxLens,
truncScheme="whole germ powers")
maxLens = [1, 2]
elgstLists2 = stdlists.make_elgst_lists(
self.opLabels,
self.germs,
maxLens,
truncScheme="whole germ powers",
nest=False,
includeLGST=False)
elgstLists2b = stdlists.make_elgst_lists(
std1Q_XY.target_model(),
self.germs,
maxLens,
truncScheme="whole germ powers",
nest=False,
includeLGST=False) # with a Model as first arg
# TODO assert correctness
def test_elgst_experiment_list(self):
elgstExpLists = stdlists.make_elgst_experiment_list(
self.opLabels, self.germs, [1, 2], truncScheme="whole germ powers")
# TODO assert correctness
def test_elgst_lists_structs_raises_on_bad_scheme(self):
with self.assertRaises(ValueError):
stdlists.make_elgst_lists(self.opLabels,
self.germs, [1, 2],
truncScheme="foobar")
def test_construction_raises_on_op_sequence_mismatch(self):
DS_bad = DataSet(outcomeLabels=['0', '1']) # order ok...
DS_bad.add_count_dict(('Gx',), {'0': 10, '1': 90})
DS_bad.done_adding_data()
with self.assertRaises(ValueError):
dc.DataComparator([self.DS_0, DS_bad])
def test_construct_empty_dataset(self):
dsEmpty = DataSet(outcomeLabels=['0', '1'])
dsEmpty.done_adding_data()