mirror of
https://github.com/Ichinga-Samuel/aiomql.git
synced 2026-08-09 10:17:46 +00:00
Ichinga Samuel
45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
import asyncio
|
|
|
|
import pytest
|
|
|
|
from aiomql.lib.result import Result
|
|
from aiomql.core.models import OrderSendResult
|
|
|
|
|
|
class TestResult:
|
|
|
|
@pytest.fixture(scope='class')
|
|
def parameters(self):
|
|
return {'name': 'test_trades', 'ema': 20, 'rsi': 14}
|
|
|
|
@pytest.fixture(scope='function')
|
|
async def order_results(self, mt, sell_order, buy_order, parameters):
|
|
res1 = await mt.order_send(sell_order)
|
|
res2 = await mt.order_send(buy_order)
|
|
res1 = Result(result=OrderSendResult(**res1._asdict()), parameters=parameters)
|
|
res2 = Result(result=OrderSendResult(**res2._asdict()), parameters=parameters)
|
|
return res1, res2
|
|
|
|
async def test_get_data(self, order_results):
|
|
res1, res2 = order_results
|
|
data1 = res1.get_data()
|
|
data2 = res2.get_data()
|
|
assert data1['actual_profit'] == data2['actual_profit'] == 0
|
|
assert data1['closed'] == data2['closed'] == False
|
|
assert data1['win'] == data2['win'] == False
|
|
|
|
|
|
async def test_csv(self, order_results):
|
|
res1, res2 = order_results
|
|
await asyncio.gather(res1.save(), res2.save())
|
|
assert res1.config.records_dir.exists()
|
|
record = res1.config.records_dir / f"{res1.name}.csv"
|
|
assert record.exists()
|
|
|
|
async def test_json(self, order_results):
|
|
res1, res2 = order_results
|
|
await asyncio.gather(res1.save(trade_record_mode='json'), res2.save(trade_record_mode='json'))
|
|
assert res1.config.records_dir.exists()
|
|
record = res1.config.records_dir / f"{res1.name}.json"
|
|
assert record.exists()
|