From 5b97864e8d3d458c39375bb3f08ed7d4e1ca088a Mon Sep 17 00:00:00 2001
From: Pablo Martin <pablo.martin@lolamarket.com>
Date: Wed, 29 Mar 2023 17:18:41 +0200
Subject: [PATCH] Signature for new task

---
 lolafect/data_testing.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/lolafect/data_testing.py b/lolafect/data_testing.py
index 4216d4b..863d6fe 100644
--- a/lolafect/data_testing.py
+++ b/lolafect/data_testing.py
@@ -60,6 +60,15 @@ def run_data_test_on_mysql(
 
     return results
 
+@task()
+def run_data_test_on_trino(
+    name: str,
+    trino_credentials: dict,
+    query: str,
+    expectation_configurations: List[ExpectationConfiguration],
+    great_expectations_s3_bucket: str = DEFAULT_GREAT_EXPECTATIONS_S3_BUCKET,
+) -> dict:
+    raise NotImplementedError("WIP")
 
 def _create_in_memory_data_context_for_mysql(
     mysql_credentials: dict,