flake8 on tests

This commit is contained in:
Jean-Baptiste LE STANG 2018-01-02 13:46:16 +01:00
parent 5344b711ea
commit 3f65fc014e

View File

@ -18,7 +18,7 @@ def whitelist_conf():
], ],
"pair_blacklist": [ "pair_blacklist": [
"BTC_BLK" "BTC_BLK"
], ],
}, },
} }
@ -36,23 +36,25 @@ def get_health():
'IsActive': True, 'IsActive': True,
'BaseVolume': 4096 'BaseVolume': 4096
} }
] ]
def get_health_empty(): def get_health_empty():
return [] return []
def test_refresh_market_pair_not_in_whitelist(mocker): def test_refresh_market_pair_not_in_whitelist(mocker):
conf = whitelist_conf() conf = whitelist_conf()
mocker.patch.dict('freqtrade.main._CONF', conf) mocker.patch.dict('freqtrade.main._CONF', conf)
mocker.patch.multiple('freqtrade.main.exchange', mocker.patch.multiple('freqtrade.main.exchange',
get_wallet_health=get_health) get_wallet_health=get_health)
refreshedwhitelist = refresh_whitelist(conf['exchange']['pair_whitelist']+ ['BTC_XXX']) refreshedwhitelist = refresh_whitelist(conf['exchange']['pair_whitelist'] + ['BTC_XXX'])
# List ordered by BaseVolume # List ordered by BaseVolume
whitelist = ['BTC_ETH', 'BTC_TKN'] whitelist = ['BTC_ETH', 'BTC_TKN']
# Ensure all except those in whitelist are removed # Ensure all except those in whitelist are removed
assert whitelist == refreshedwhitelist assert whitelist == refreshedwhitelist
def test_refresh_whitelist(mocker): def test_refresh_whitelist(mocker):
conf = whitelist_conf() conf = whitelist_conf()
mocker.patch.dict('freqtrade.main._CONF', conf) mocker.patch.dict('freqtrade.main._CONF', conf)