diff --git a/lolafect/data_testing.py b/lolafect/data_testing.py index 4e4f45e..1d0dc11 100644 --- a/lolafect/data_testing.py +++ b/lolafect/data_testing.py @@ -23,7 +23,7 @@ def run_data_test_on_mysql( """ Validate a query and an expectation suite against a given MySQL server. - :param name: a name for the data test. + :param name: a unique name for the data test. :param mysql_credentials: credentials for the MySQL instance. :param query: the query to test against. :param expectations: the expectations on the dataset. @@ -60,7 +60,7 @@ def create_in_memory_data_context( data_context = BaseDataContext( project_config=DataContextConfig( datasources={ - "HARCODED": DatasourceConfig( # f"{mysql_credentials['host']}:{mysql_credentials['port']}": DatasourceConfig( + f"{mysql_credentials['host']}:{mysql_credentials['port']}": DatasourceConfig( class_name="Datasource", execution_engine={ "class_name": "SqlAlchemyExecutionEngine", @@ -131,7 +131,7 @@ def create_checkpoint( "validations": [ { "batch_request": { - "datasource_name": "HARCODED", + "datasource_name": f"{mysql_credentials['host']}:{mysql_credentials['port']}", "data_connector_name": "default_runtime_data_connector_name", "data_asset_name": f"{expectation_suite_name}_validation_query", "runtime_parameters": {"query": query_for_checkpoint},