Add to_table() in integration tests

Signed-off-by: Matt Delacour <matt.delacour@shopify.com>
feast-dev · Jun 29, 2021 · c378307 · c378307
1 parent fa5551b
commit c378307
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/sdk/python/tests/test_historical_retrieval.py b/sdk/python/tests/test_historical_retrieval.py
@@ -461,6 +461,11 @@ def test_historical_features_from_bigquery_sources(
             check_dtype=False,
         )
 
+        table_from_sql_entities = job_from_sql.to_table()
+        assert_frame_equal(
+            actual_df_from_sql_entities, table_from_sql_entities.to_pandas()
+        )
+
         timestamp_column = (
             "e_ts"
             if infer_event_timestamp_col
@@ -541,3 +546,7 @@ def test_historical_features_from_bigquery_sources(
             .reset_index(drop=True),
             check_dtype=False,
         )
+        table_from_df_entities = job_from_df.to_table()
+        assert_frame_equal(
+            actual_df_from_df_entities, table_from_df_entities.to_pandas()
+        )