From 65c617ed94e39cd3d12d58c5d6b35f894f41f892 Mon Sep 17 00:00:00 2001 From: karlicoss Date: Fri, 10 Nov 2023 02:00:37 +0000 Subject: [PATCH] my.emfit: add missing properties to fake data generator --- my/emfit/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/my/emfit/__init__.py b/my/emfit/__init__.py index 1ec3341..30b693c 100644 --- a/my/emfit/__init__.py +++ b/my/emfit/__init__.py @@ -170,6 +170,8 @@ def fake_data(nights: int = 500) -> Iterator: from my.core.cfg import tmp_config from tempfile import TemporaryDirectory + import pytz + with TemporaryDirectory() as td: tdir = Path(td) gen = dal.FakeData() @@ -178,6 +180,8 @@ def fake_data(nights: int = 500) -> Iterator: class override: class emfit: export_path = tdir + excluded_sids = () + timezone = pytz.timezone('Europe/London') # meh with tmp_config(modules=__name__, config=override) as cfg: yield cfg