def execute(self, context):
"""
This method check the quality of the data given the input test cases indicated in the sql_test_cases
dictionary.
"""
bigquery = BigQueryHook(bigquery_conn_id=self.conn_id)
found_errors = []
for query, expected_result in self.sql_test_cases.items():
records = bigquery.run_query(sql=query)
if len(records) < 1 or records[0][0] != expected_result:
found_errors.append(query)
if len(found_errors) > 0:
raise ValueError(
f"The following query test cases were not successful {found_errors}"
)
self.log.info('DataQualityOperator has been executed')
