def setUp(self): testfile = os.path.join(curdir, 'lattice/test_0.lat') out1file = os.path.join(curdir, 'lattice/out1_0.lat') out2file = os.path.join(curdir, 'lattice/out2_0.lat') self.testfile = make_latfile(testfile) self.out1file = make_latfile(out1file) self.out2file = make_latfile(out2file) with open(self.testfile, 'rb') as f: self.m = Machine(f) s0 = self.m.allocState({}) self.r = self.m.propagate(s0, 0, len(self.m), range(len(self.m))) self.s = s0 k = [ 'beta', 'bg', 'gamma', 'IonEk', 'IonEs', 'IonQ', 'IonW', 'IonZ', 'phis', 'SampleIonK' ] self.keys = [i for i in k] self.ref_keys = ['ref_{}'.format(i) for i in k] + ['pos'] self.keys_ps = ['moment0', 'moment0_env', 'moment0_rms', 'moment1'] with open(self.out1file, 'rb') as f: self.fout1_str = f.read().strip().decode() with open(self.out2file, 'rb') as f: self.fout2_str = f.read().strip().decode() self.latfile0 = os.path.join(curdir, 'lattice/test_1.lat') self.latfile1 = os.path.join(curdir, 'lattice/out1_1.lat') self.latfile2 = os.path.join(curdir, 'lattice/out2_1.lat') self.testcfile = os.path.join(curdir, 'lattice/test_c.lat') self.out3file = os.path.join(curdir, 'lattice/out3.lat') with open(self.out3file, 'rb') as f: self.fout3_str = f.read().strip().decode() with open(self.testcfile, 'rb') as f: self.mc = Machine(f) self.foutc_str = f.read().strip().decode() out4file = os.path.join(curdir, 'lattice/out4_0.lat') self.out4file = make_latfile(out4file) with open(self.out4file, 'rb') as f: self.fout4_str = f.read().strip().decode() self.latfile4 = os.path.join(curdir, 'lattice/out4_1.lat')
def setUp(self): latfile = os.path.join(curdir, 'lattice/test_0.lat') self.latfile = make_latfile(latfile) with open(self.latfile, 'rb') as f: self.m = Machine(f)
def setUp(self): latfile = os.path.join(curdir, 'lattice/test_0.lat') self.latfile = make_latfile(latfile) self.inslat = os.path.join(curdir, 'data/inslat.out')
def setUp(self): testfile = os.path.join(curdir, 'lattice/test_0.lat') self.testfile = make_latfile(testfile)