Skip to content

Commit

Permalink
FIX: Use timezone import.
Browse files Browse the repository at this point in the history
  • Loading branch information
zssherman committed Apr 11, 2024
1 parent 1def22a commit 7ac400e
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
6 changes: 3 additions & 3 deletions tests/qc/test_qcfilter.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
import copy
import datetime
from datetime import datetime, timezone

import dask.array as da
import numpy as np
Expand Down Expand Up @@ -390,12 +390,12 @@ def test_qc_speed():
coords={'time': time},
)

start = datetime.datetime.now(datetime.UTC)
start = datetime.now(timezone.utc)
for name, var in noisy_data_mapping.items():
failed_qc = var > 0.75 # Consider data above 0.75 as bad. Negligible time here.
ds.qcfilter.add_test(name, index=failed_qc, test_meaning='Value above threshold')

time_diff = datetime.datetime.now(datetime.UTC) - start
time_diff = datetime.now(timezone.utc) - start
assert time_diff.seconds <= 4


Expand Down
4 changes: 2 additions & 2 deletions tests/utils/test_datetime_utils.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from datetime import datetime
from datetime import datetime, timezone

import numpy as np
import pandas as pd
Expand All @@ -15,7 +15,7 @@ def test_dates_between():
answer = np.arange(start_string, end_string, dtype='datetime64[D]')
answer = np.append(answer, answer[-1] + 1)
answer = answer.astype('datetime64[s]').astype(int)
answer = [datetime.fromtimestamp(ii, datetime.UTC) for ii in answer]
answer = [datetime.fromtimestamp(ii, timezone.utc) for ii in answer]

assert date_list == answer

Expand Down

0 comments on commit 7ac400e

Please sign in to comment.