def main(unused_argv):
cfg = config.Config(FLAGS.config,
mode=FLAGS.mode,
env_path=FLAGS.dotenv_path)
store_factory = db_store.create_store_factory_for_sqlite_db(cfg)
store = store_factory.create()
populate_store_fake(store)
def test_load_test_data():
"""Ensure consitency between test data used in plots and BD schema
e.g. between load_test_data and db.get_bed_counts
"""
# TODO: cache generation of in memory DB with fake data into a pytest fixture
df = load_test_data()['bedcounts']
# duplicate columns with "date"
del df['datetime']
store = StoreFactory(create_engine("sqlite:///:memory:")).create()
populate_store_fake(store)
df_db = to_pandas(store.get_bed_counts())
df_db = normalize_colum_names(df_db, 'bedcounts')
# All bedcount columns in tests data used for plots must belong to the DB schema
missing_columns = set(df.columns).difference(df_db.columns)
assert not missing_columns
def setUp(self):
self.store_factory = StoreFactory(
create_engine("sqlite:///:memory:", echo=False))
self.store = self.store_factory.create()
populate_store_fake(self.store)
def get_app(self):
populate_store_fake(self.db)
self.server.dataset.db = self.db
return self.server.make_app()
def fake_db():
store_factory = store.StoreFactory(
sqla.create_engine("sqlite:///:memory:", echo=False))
db = store_factory.create()
populate_store_fake(db)
return db
