Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Infer min and max timestamps from entity_df to limit data read from BQ source #1665

Merged
merged 11 commits into from
Jul 8, 2021
Merged
Prev Previous commit
Next Next commit
Explicitly set retrieved timestamps to same timezone when testing min…
…/max timestamp inference

Signed-off-by: Mwad22 <51929507+Mwad22@users.noreply.github.com>
  • Loading branch information
Mwad22 committed Jul 6, 2021
commit 12daf0148ab5bfd7f4c0fd380e92eea19826fa46
4 changes: 2 additions & 2 deletions sdk/python/tests/test_historical_retrieval.py
Original file line number Diff line number Diff line change
Expand Up @@ -578,5 +578,5 @@ def test_timestamp_bound_inference_from_entity_df_using_bigquery():
client, str(table.reference), "e_ts"
)

assert min_timestamp is min(entity_df["e_ts"])
assert max_timestamp is max(entity_df["e_ts"])
assert min_timestamp.astimezone("UTC") is min(entity_df["e_ts"]).astimezone("UTC")
assert max_timestamp.astimezone("UTC") is max(entity_df["e_ts"]).astimezone("UTC")