Ejemplo n.º 1
0
    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')
Ejemplo n.º 4
0
 def setUp(self):
     testfile = os.path.join(curdir, 'lattice/test_0.lat')
     self.testfile = make_latfile(testfile)