hack for bq tests

carbonfact · Nov 9, 2023 · da817ff · da817ff
1 parent 10da16f
commit da817ff
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/lea/app/test.py b/lea/app/test.py
@@ -31,7 +31,6 @@ def test(
         view_columns = columns.query(f"view_name == '{client._make_view_path(view)}'")[
             "column"
         ].tolist()
-
         for test in client.yield_unit_tests(view=view, view_columns=view_columns):
             assertion_tests.append(test)
     console.log(f"Found {len(assertion_tests):,d} assertion tests")

diff --git a/lea/clients/bigquery.py b/lea/clients/bigquery.py
@@ -133,7 +133,10 @@ def get_columns(self) -> pd.DataFrame:
         FROM {self.dataset_name}.INFORMATION_SCHEMA.COLUMNS
         """
         view = lea.views.GenericSQLView(schema=None, name=None, query=query, sqlglot_dialect=self.sqlglot_dialect)
-        return self._load_sql_view(view)
+        columns = self._load_sql_view(view)
+        # HACK
+        columns['view_name'] = columns['view_name'].apply(lambda x: f"{self.dataset_name}.{x}")
+        return columns
 
     def _make_view_path(self, view: lea.views.View) -> str:
         return f"{self.dataset_name}.{lea._SEP.join(view.key)}"

diff --git a/lea/clients/duckdb.py b/lea/clients/duckdb.py
@@ -89,7 +89,7 @@ def get_tables(self):
     def get_columns(self) -> pd.DataFrame:
         query = """
         SELECT
-             table_schema || '.' || table_name AS view_name,
+            table_schema || '.' || table_name AS view_name,
             column_name AS column,
             data_type AS type
         FROM information_schema.columns