gavindiaz/aiomql
1
0
Fork 0
mirror of https://github.com/Ichinga-Samuel/aiomql.git synced 2026-08-18 06:18:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

v4

Browse Source
This commit is contained in:
Ichinga Samuel
2024-11-04 00:32:51 +01:00
parent 2c46e18528
commit d65a8db1a2
85 changed files with 5071 additions and 2512 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/live/unit/test_get_data.py
+17 -10
View File
@@ -3,7 +3,7 @@ from datetime import datetime, UTC
import pytest
from aiomql.contrib.backtesting.get_data import GetData
from aiomql.core.backtesting.get_data import GetData
from aiomql.core.constants import TimeFrame
@@ -12,12 +12,17 @@ class TestGetData:
def setup_class(cls):
cls.start = datetime(2024, 2, 1, tzinfo=UTC)
cls.end = datetime(2024, 2, 2, tzinfo=UTC)
cls.symbols = ['BTCUSD', "ETHUSD"]
cls.symbols = ["BTCUSD", "ETHUSD"]
cls.timeframes = [TimeFrame.H1, TimeFrame.H2]
cls.g_data = GetData(start=cls.start, end=cls.end, symbols=cls.symbols, timeframes=cls.timeframes,
name='test_data')
cls.g_data = GetData(
start=cls.start,
end=cls.end,
symbols=cls.symbols,
timeframes=cls.timeframes,
name="test_data",
)
@pytest.fixture(scope='class', autouse=True)
@pytest.fixture(scope="class", autouse=True)
async def get_data(self):
await self.g_data.get_data()
self.g_data.save_data()
@@ -27,9 +32,11 @@ class TestGetData:
assert self.g_data.end == self.end
assert self.g_data.symbols == set(self.symbols)
assert self.g_data.timeframes == set(self.timeframes)
assert self.g_data.name == 'test_data'
assert self.g_data.name == "test_data"
assert self.g_data.range == range(int((self.end - self.start).total_seconds()))
assert self.g_data.span == range(int(self.start.timestamp()), int(self.end.timestamp()))
assert self.g_data.span == range(
int(self.start.timestamp()), int(self.end.timestamp())
)
async def test_get_data(self):
assert self.g_data.data.fully_loaded is True
@@ -37,12 +44,12 @@ class TestGetData:
assert len(self.g_data.data.symbols.keys()) == 2
async def test_save_data(self):
file = Path(self.g_data.config.backtest_dir / 'test_data.pkl')
file = Path(self.g_data.config.backtest_dir / "test_data.pkl")
assert file.exists()
async def test_load_data(self):
data = GetData.load_data(name='tests/live/backtesting/test_data.pkl')
assert data.name == 'test_data'
data = GetData.load_data(name="tests/live/backtesting/test_data.pkl")
assert data.name == "test_data"
assert data.fully_loaded is True
assert len(data.ticks.keys()) == 2
assert len(data.symbols.keys()) == 2