chore: format output

etonlels · May 8, 2024 · 91589b9 · 91589b9
1 parent c720b14
commit 91589b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/core/engine_adapter/test_integration.py b/tests/core/engine_adapter/test_integration.py
@@ -173,7 +173,7 @@ def get_current_data(self, table: exp.Table) -> pd.DataFrame:
         df = self.engine_adapter.fetchdf(exp.select("*").from_(table), quote_identifiers=True)
         if self.dialect == "snowflake" and "id" in df.columns:
             df["id"] = df["id"].apply(lambda x: x if pd.isna(x) else int(x))
-        return df
+        return self._format_df(df)
 
     def compare_with_current(self, table: exp.Table, expected: pd.DataFrame) -> None:
         compare_dataframes(