Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions src/plots/elexon_plots.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
from typing import Callable, List, Optional, Tuple, Union
import pandas as pd
from datetime import datetime, date, timedelta
from datetime import datetime, date, timedelta, UTC
from plotly import graph_objects as go
import streamlit as st
from elexonpy.api_client import ApiClient
Expand Down Expand Up @@ -132,7 +132,7 @@ def determine_start_and_end_datetimes(
- start_datetime_utc: datetime object in UTC
- end_datetime_utc: datetime object in UTC
"""
now = datetime.utcnow()
now = datetime.now(UTC)

# Determine start_datetime_utc
if start_datetimes:
Expand Down
4 changes: 2 additions & 2 deletions tests/test_elexon_plot.py
Original file line number Diff line number Diff line change
@@ -1,15 +1,15 @@
from unittest.mock import Mock, patch
import pandas as pd
import pytest
from datetime import datetime
from datetime import datetime, UTC
from plotly import graph_objects as go
from plots.elexon_plots import add_elexon_plot, determine_start_and_end_datetimes, fetch_forecast_data
from elexonpy.api_client import ApiClient
from elexonpy.api.generation_forecast_api import GenerationForecastApi

def test_determine_start_and_end_datetimes_no_input():
# Test with no input
now = datetime.utcnow()
now = datetime.now(UTC)
start, end = determine_start_and_end_datetimes([], [])
assert start < now, "Start time should be before current time."
assert end > start, "End time should be after start time."
Expand Down