Parametrize tests

This commit is contained in:
Matthias 2021-08-09 11:35:27 +02:00
parent de4b0d1f7c
commit 599ae15460

View File

@ -1,65 +1,37 @@
from decimal import Decimal
from math import isclose
import pytest
from freqtrade.leverage import interest
# from freqtrade.exceptions import OperationalException
binance = "binance"
kraken = "kraken"
ftx = "ftx"
other = "bittrex"
def test_interest():
borrowed = Decimal(60.0)
interest_rate = Decimal(0.0005)
interest_rate_2 = Decimal(0.00025)
ten_mins = Decimal(1/6)
five_hours = Decimal(5.0)
twentyfive_hours = Decimal(25.0)
# Binance
assert float(interest(
exchange_name=binance,
borrowed=borrowed,
rate=interest_rate,
hours=ten_mins
)) == 0.00125
assert float(interest(
exchange_name=binance,
borrowed=borrowed,
rate=interest_rate_2,
hours=five_hours
)) == 0.003125
@pytest.mark.parametrize('exchange,interest_rate,hours,expected', [
('binance', 0.0005, ten_mins, 0.00125),
('binance', 0.00025, ten_mins, 0.000625),
('binance', 0.00025, five_hours, 0.003125),
('binance', 0.00025, twentyfive_hours, 0.015625),
# Kraken
assert float(interest(
exchange_name=kraken,
borrowed=borrowed,
rate=interest_rate,
hours=ten_mins
)) == 0.06
assert float(interest(
exchange_name=kraken,
borrowed=borrowed,
rate=interest_rate_2,
hours=five_hours
)) == 0.045
('kraken', 0.0005, ten_mins, 0.06),
('kraken', 0.00025, ten_mins, 0.03),
('kraken', 0.00025, five_hours, 0.045),
('kraken', 0.00025, twentyfive_hours, 0.12),
# FTX
# TODO-lev
# assert float(interest(
# exchange_name=ftx,
# borrowed=borrowed,
# rate=interest_rate,
# hours=ten_mins
# )) == 0.00125
# TODO-lev: - implement FTX tests
# ('ftx', Decimal(0.0005), ten_mins, 0.06),
# ('ftx', Decimal(0.0005), five_hours, 0.045),
])
def test_interest(exchange, interest_rate, hours, expected):
borrowed = Decimal(60.0)
# assert float(interest(
# exchange_name=ftx,
# borrowed=borrowed,
# rate=interest_rate_2,
# hours=five_hours
# )) == 0.003125
assert isclose(interest(
exchange_name=exchange,
borrowed=borrowed,
rate=Decimal(interest_rate),
hours=hours
), expected)