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()
