def setUp(self): """Create a test file entry.""" self.file = File.from_path(__file__) self.conn = engine.connect() self.trans = self.conn.begin() self.session = session_factory(bind=self.conn) self.session.add(self.file) self.session.commit()
def setUp(self): """Set up the test job and test (related) file.""" self.job = Job(name="test") self.file = File(name="test", uid="test_uid"*8, ext="nul") self.conn = engine.connect() self.trans = self.conn.begin() self.session = session_factory(bind=self.conn) self.job.files.append(self.file) self.session.add(self.job) self.session.commit()
def get_by_symbol(cls, symbol): """Get all isotopes with a given element symbol.""" return session_factory().query(cls).filter(cls.symbol == symbol).all()
def get_by_strid(cls, strid): """Get an isotope using a string id.""" return session_factory().query(cls).filter(cls.strid == strid).one()