mirror of
https://github.com/Ichinga-Samuel/aiomql.git
synced 2026-08-03 15:27:44 +00:00
Ichinga Samuel
43 lines
1.6 KiB
Python
43 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()
|