Forcing UTC datetimes

bytewax · Sep 21, 2022 · 0cf8971 · 0cf8971
1 parent 14ea8ac
commit 0cf8971
Show file tree

Hide file tree

Showing 10 changed files with 313 additions and 795 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -13,10 +13,9 @@ python-source = "pysrc"
 [dependencies]
 bincode = { version = "1.3.3" }
 chrono = { version = "0.4", features = [ "serde" ] }
-chrono-tz = { version = "0.6", features = ["serde"] }
 futures = { version = "0.3.21" }
 log = { version = "0.4" }
-pyo3 = { git = "https://github.com/psykopear/pyo3", branch = "chrono", version = "0.17" }
+pyo3 = { git = "https://github.com/psykopear/pyo3", branch = "chrono", version = "0.17", features = ["macros", "chrono"] }
 # pyo3-log = { version = "0.7" }
 rdkafka = { version = "0.28.0", features = [ "cmake-build" ] }
 scopeguard = { version = "1.1.0" }
@@ -28,7 +27,7 @@ timely = { version = "0.12.0", features = [ "bincode" ] }
 tokio = { version = "1.20.1", features = [ "full" ] }
 
 [dev-dependencies]
-pyo3 = { git = "https://github.com/psykopear/pyo3", branch = "chrono", version = "0.17", default-features = false }
+pyo3 = { git = "https://github.com/psykopear/pyo3", branch = "chrono", version = "0.17", default-features = false, features = ["macros", "chrono"] }
 
 [features]
 extension-module = ["pyo3/extension-module"]

diff --git a/pytests/test_datetime.py b/pytests/test_datetime.py
diff --git a/pytests/test_window.py b/pytests/test_window.py
@@ -1,5 +1,5 @@
 from collections import defaultdict
-from datetime import timedelta, datetime
+from datetime import timedelta, datetime, timezone
 
 from bytewax.dataflow import Dataflow
 from bytewax.execution import run_main
@@ -17,7 +17,8 @@ def gen():
 
     flow.input("inp", TestingInputConfig(gen()))
 
-    start_at = datetime(2022, 1, 1)
+    # Only utc datetime is accepted as the `start_at` parameter
+    start_at = datetime(2022, 1, 1, tzinfo=timezone.utc)
     # This will result in times for events of +0, +4, +8, +12.
     clock_config = TestingClockConfig(item_incr=timedelta(seconds=4), start_at=start_at)
     # And since the window is +10, we should get a window with value
@@ -61,7 +62,8 @@ def count(results, event):
         results[event["type"]] += 1
         return results
 
-    start_at = datetime(2022, 1, 1)
+    # Only utc datetime is accepted as the `start_at` parameter
+    start_at = datetime(2022, 1, 1, tzinfo=timezone.utc)
     # This will result in times for events of +0, +4, +8, +12.
     clock_config = TestingClockConfig(
         item_incr=timedelta(seconds=4), start_at=start_at