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