CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2025-11-30 00:53:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

pylint fixes

Browse Source
This commit is contained in:
gcarq
2017-09-08 23:10:22 +02:00
parent 689cd11a6c
commit 996beae770
9 changed files with 133 additions and 121 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
test/test_main.py
View File

@@ -1,12 +1,11 @@
import unittest
from unittest.mock import patch, MagicMock
import os
from jsonschema import validate
import exchange
from main import create_trade, handle_trade, close_trade_if_fulfilled, init
from misc import conf_schema
from misc import CONF_SCHEMA
from persistence import Trade
@@ -43,7 +42,7 @@ class TestMain(unittest.TestCase):
}
def test_1_create_trade(self):
with patch.dict('main._conf', self.conf):
with patch.dict('main._CONF', self.conf):
with patch('main.get_buy_signal', side_effect=lambda _: True) as buy_signal:
with patch.multiple('main.telegram', init=MagicMock(), send_msg=MagicMock()):
with patch.multiple('main.exchange',
@@ -68,7 +67,7 @@ class TestMain(unittest.TestCase):
buy_signal.assert_called_once_with('BTC_ETH')
def test_2_handle_trade(self):
with patch.dict('main._conf', self.conf):
with patch.dict('main._CONF', self.conf):
with patch.multiple('main.telegram', init=MagicMock(), send_msg=MagicMock()):
with patch.multiple('main.exchange',
get_ticker=MagicMock(return_value={
@@ -86,7 +85,7 @@ class TestMain(unittest.TestCase):
self.assertEqual(trade.open_order_id, 'dry_run')
def test_3_close_trade(self):
with patch.dict('main._conf', self.conf):
with patch.dict('main._CONF', self.conf):
trade = Trade.query.filter(Trade.is_open.is_(True)).first()
self.assertTrue(trade)
@@ -99,7 +98,7 @@ class TestMain(unittest.TestCase):
@classmethod
def setUpClass(cls):
validate(cls.conf, conf_schema)
validate(cls.conf, CONF_SCHEMA)
if __name__ == '__main__':