def test_sqlalchemysource_templating(sqlitedb_engine):
datasource = SqlAlchemyDatasource(engine=sqlitedb_engine)
generator = datasource.get_generator()
generator.add_query("test", "select 'cat' as ${col_name};")
df = datasource.get_batch("test", col_name="animal_name")
res = df.expect_column_to_exist("animal_name")
assert res["success"] == True
def test_sqlalchemy_source_templating(sqlitedb_engine):
datasource = SqlAlchemyDatasource(
engine=sqlitedb_engine,
generators={"foo": {
"class_name": "QueryGenerator"
}})
generator = datasource.get_generator("foo")
generator.add_query("test", "select 'cat' as ${col_name};")
df = datasource.get_batch(
"test", "my_suite",
generator.yield_batch_kwargs("test", col_name="animal_name"))
res = df.expect_column_to_exist("animal_name")
assert res["success"] is True
def test_sqlalchemy_source_templating(sqlitedb_engine):
datasource = SqlAlchemyDatasource(engine=sqlitedb_engine, generators={
"foo": {
"class_name": "QueryBatchKwargsGenerator"
}
})
generator = datasource.get_generator("foo")
generator.add_query("test", "select 'cat' as ${col_name};")
batch = datasource.get_batch(generator.build_batch_kwargs("test", query_parameters={'col_name': "animal_name"}))
dataset = Validator(batch, expectation_suite=ExpectationSuite("test"), expectation_engine=SqlAlchemyDataset).get_dataset()
res = dataset.expect_column_to_exist("animal_name")
assert res.success is True
res = dataset.expect_column_values_to_be_in_set('animal_name', ['cat'])
assert res.success is True
