def test_sqlalchemy_source_templating(sqlitedb_engine, empty_data_context):
context: DataContext = empty_data_context
datasource = SqlAlchemyDatasource(
engine=sqlitedb_engine,
batch_kwargs_generators={
"foo": {
"class_name": "QueryBatchKwargsGenerator"
}
},
)
generator = datasource.get_batch_kwargs_generator("foo")
generator.add_query(data_asset_name="test",
query="select 'cat' as ${col_name};")
batch = datasource.get_batch(
generator.build_batch_kwargs(
"test", query_parameters={"col_name": "animal_name"}))
dataset = BridgeValidator(
batch,
expectation_suite=ExpectationSuite("test", data_context=context),
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
def test_sqlalchemy_source_templating(sqlitedb_engine):
datasource = SqlAlchemyDatasource(engine=sqlitedb_engine,
batch_kwargs_generators={
"foo": {
"class_name":
"QueryBatchKwargsGenerator"
}
})
generator = datasource.get_batch_kwargs_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
