def create(self, test):
session = test.session()
with session.begin() as t:
for name, typ, ccy in zip(self.inames, self.itypes, self.iccys):
t.add(Instrument(name=name, type=typ, ccy=ccy))
for name in self.gnames:
t.add(Group(name=name))
for name, ccy in zip(self.inames, self.iccys):
t.add(Fund(name=name, ccy=ccy))
yield t.on_result
iall = yield test.session().query(Instrument).load_only('id').all()
fall = yield test.session().query(Fund).load_only('id').all()
with session.begin() as t:
for i in iall:
t.add(ObjectAnalytics(model_type=Instrument, object_id=i.id))
for i in fall:
t.add(ObjectAnalytics(model_type=Fund, object_id=i.id))
yield t.on_result
obj_len = self.size[1]
groups = yield session.query(Group).all()
objs = yield session.query(ObjectAnalytics).all()
groups = self.populate('choice', obj_len, choice_from=groups)
objs = self.populate('choice', obj_len, choice_from=objs)
with test.session().begin() as t:
for g, o in zip(groups, objs):
t.add(AnalyticData(group=g, object=o))
yield t.on_result
def setUp(self):
'''Create Instruments and Funds commiting at the end for speed'''
orm = self.orm
orm.register(Instrument)
orm.register(Fund)
orm.register(Position)
orm.register(PortfolioView)
orm.register(UserDefaultView)
for name,typ,ccy in izip(inst_names,inst_types,inst_ccys):
Instrument(name = name, type = typ, ccy = ccy).save(False)
Instrument.commit()
for name,ccy in izip(fund_names,fund_ccys):
Fund(name = name, ccy = ccy).save(False)
Fund.commit()
