def test_set_bmstate(self): fm_none = ModelFlame() self.assertIsNone(fm_none.bmstate) with open(self.testfile, 'rb') as f: m = Machine(f) s = m.allocState({}) m.propagate(s, 0, 1) fm_none.bmstate = s compare_mstates(self, fm_none.bmstate, s)
def test_run_5(self): """ test_run_5: using BeamState object """ latfile = self.testfile with open(latfile, 'rb') as f: m0 = Machine(f) ms = BeamState(machine=m0) fm = ModelFlame() fm.bmstate = ms fm.machine = m0 obs = fm.get_index_by_type(type='bpm')['bpm'] r,s = fm.run(monitor=obs) s0 = m0.allocState({}) m0.propagate(s0, 0, 1) r0 = m0.propagate(s0, 1, len(m0), observe=obs) rs0 = [ts for (ti,ts) in r0] rs = [ts for (ti,ts) in r] for (is1, is2) in zip(rs0, rs): compare_mstates(self, is1, is2) compare_mstates(self, s, s0)