From 02bf00d1ff5c73b6c82aa207577fae05caa4c38d Mon Sep 17 00:00:00 2001 From: peterdudfield Date: Wed, 13 Nov 2024 07:19:13 +0000 Subject: [PATCH] lint --- tests/conftest.py | 3 ++- tests/test_adjuster.py | 5 +++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index a902141..b61784c 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -6,6 +6,7 @@ import logging import os import random +from uuid import uuid4 import numpy as np import pandas as pd @@ -17,7 +18,6 @@ from pvsite_datamodel.sqlmodels import Base, GenerationSQL, ForecastSQL, ForecastValueSQL, SiteSQL from sqlalchemy import create_engine from testcontainers.postgres import PostgresContainer -from uuid import uuid4 log = logging.getLogger(__name__) @@ -202,6 +202,7 @@ def forecast_values(): @pytest.fixture() def forecasts(db_session, sites): + """Make fake forecasts""" init_timestamp = pd.Timestamp(dt.datetime.now(tz=None)).floor(dt.timedelta(minutes=15)) n = 24 * 4 # 24 hours of readings of 15 diff --git a/tests/test_adjuster.py b/tests/test_adjuster.py index 163c953..ec43b08 100644 --- a/tests/test_adjuster.py +++ b/tests/test_adjuster.py @@ -1,7 +1,8 @@ -from india_forecast_app.adjuster import get_me_values, adjust_forecast_with_adjuster +from datetime import datetime + import pandas as pd -from datetime import datetime +from india_forecast_app.adjuster import get_me_values, adjust_forecast_with_adjuster def test_get_me_values_no_values(db_session, sites):