def __init__(self, engine, name, model, fact_table=None):
self.name = name
if not isinstance(model, Model):
model = Model(model)
self._fact_table = fact_table
self.model = model
self.engine = engine
self.meta = MetaData(bind=engine)
def model(self):
if self._model is None:
if not self.fact_table.exists:
return
data = self.data.get('model')
if not isinstance(data, dict):
return
data['fact_table'] = self.fact_table.table_name
model = Model(data)
if not model.exists:
return
self._model = model
return self._model
def setUp(self):
super(ModelTestCase, self).setUp()
self.simple_model_data = load_json_fixture('models/simple_model.json')
self.simple_model = Model(self.simple_model_data)
def model(self):
if not hasattr(self, '_model') or self._model is None:
self._model = Model(self.spec.get('model', {}))
return self._model
