margin test_trade_close

freqtrade/persistence/models.py

@@ -610,33 +610,39 @@ class LocalTrade():
 
     def calculate_interest(self) -> float:
         # TODO-mg: Need to set other conditions because sometimes self.open_date is not defined, but why would it ever not be set
+        zero = Decimal(0.0)
         if not self.interest_rate or not (self.borrowed):
-            return 0.0
+            return zero
 
         open_date = self.open_date.replace(tzinfo=None)
         now = datetime.utcnow()
-        secPerDay = 86400
+        # sec_per_day = Decimal(86400)
-        days = ((now - open_date).total_seconds())/secPerDay or 0.0
+        sec_per_hour = Decimal(3600)
-        hours = days * 24
+        total_seconds = Decimal((now - open_date).total_seconds())
+        #days = total_seconds/sec_per_day or zero
+        hours = total_seconds/sec_per_hour or zero
 
-        rate = self.interest_rate
+        rate = Decimal(self.interest_rate)
-        borrowed = self.borrowed
+        borrowed = Decimal(self.borrowed)
+        one = Decimal(1.0)
+        twenty_four = Decimal(24.0)
+        four = Decimal(4.0)
 
         if self.exchange == 'binance':
             # Rate is per day but accrued hourly or something
             # binance: https://www.binance.com/en-AU/support/faq/360030157812
-            return borrowed * rate * max(hours, 1)/24  # TODO-mg: Is hours rounded?
+            return borrowed * rate * max(hours, one)/twenty_four  # TODO-mg: Is hours rounded?
         elif self.exchange == 'kraken':
             # https://support.kraken.com/hc/en-us/articles/206161568-What-are-the-fees-for-margin-trading-
             opening_fee = borrowed * rate
-            roll_over_fee = borrowed * rate * max(0, (hours-4)/4)
+            roll_over_fee = borrowed * rate * max(0, (hours-four)/four)
             return opening_fee + roll_over_fee
         elif self.exchange == 'binance_usdm_futures':
             # ! TODO-mg: This is incorrect, I didn't look it up
-            return borrowed * (rate/24) * max(hours, 1)
+            return borrowed * (rate/twenty_four) * max(hours, one)
         elif self.exchange == 'binance_coinm_futures':
             # ! TODO-mg: This is incorrect, I didn't look it up
-            return borrowed * (rate/24) * max(hours, 1)
+            return borrowed * (rate/twenty_four) * max(hours, one)
         else:
             # TODO-mg: make sure this breaks and can't be squelched
             raise OperationalException("Leverage not available on this exchange")

tests/conftest.py

@@ -57,9 +57,9 @@ def log_has_re(line, logs):
 
 def get_args(args):
     return Arguments(args).get_parsed_arg()
+
+
 # Source: https://stackoverflow.com/questions/29881236/how-to-mock-asyncio-coroutines
-
-
 def get_mock_coro(return_value):
     async def mock_coro(*args, **kwargs):
         return return_value

tests/test_persistence.py

@@ -196,6 +196,7 @@ def test_calc_open_close_trade_price(limit_buy_order, limit_sell_order, fee):
 
 @pytest.mark.usefixtures("init_persistence")
 def test_trade_close(limit_buy_order, limit_sell_order, fee):
+    #TODO: limit_buy_order and limit_sell_order aren't used, remove them probably
     trade = Trade(
         pair='ETH/BTC',
         stake_amount=0.001,

tests/test_persistence_margin.py

@@ -211,26 +211,60 @@ def test_calc_open_close_trade_price(limit_short_order, limit_exit_short_order,
     assert isclose(trade.calc_profit_ratio(), 0.06189996)
 
 
-# @pytest.mark.usefixtures("init_persistence")
+@pytest.mark.usefixtures("init_persistence")
-# def test_trade_close(limit_buy_order, limit_sell_order, fee):
+def test_trade_close(fee, five_hours_ago):
-#     trade = Trade(
+    """
-#         pair='ETH/BTC',
+        This trade lasts for five hours, but the one above lasted for 10 minutes
-#         stake_amount=0.001,
+        Short trade
-#         open_rate=0.01,
+        Exchange: Kraken
-#         amount=5,
+        fee: 0.25% base
-#         is_open=True,
+        interest_rate: 0.05% per 4 hours
-#         fee_open=fee.return_value,
+        open_rate: 0.02 base
-#         fee_close=fee.return_value,
+        close_rate: 0.01 base
-#         open_date=arrow.Arrow(2020, 2, 1, 15, 5, 1).datetime,
+        leverage: 3.0
-#         exchange='binance',
+        amount: 5 * 3 = 15 crypto
-#     )
+        borrowed: 15 crypto
-#     assert trade.close_profit is None
+        time-periods: 5 hours = 5/4
-#     assert trade.close_date is None
+
-#     assert trade.is_open is True
+        interest: borrowed * interest_rate * time-periods
-#     trade.close(0.02)
+                    = 15 * 0.0005 * 5/4 = 0.009375 crypto
-#     assert trade.is_open is False
+        open_value: (amount * open_rate) - (amount * open_rate * fee)
-#     assert trade.close_profit == 0.99002494
+            = (15 * 0.02) - (15 * 0.02 * 0.0025)
-#     assert trade.close_date is not None
+            = 0.29925
+        amount_closed: amount + interest = 15 + 0.009375 = 15.009375
+        close_value: (amount_closed * close_rate) + (amount_closed * close_rate * fee)
+            = (15.009375 * 0.01) + (15.009375 * 0.01 * 0.0025)
+            = 0.150468984375
+        total_profit = open_value - close_value
+            = 0.29925 - 0.150468984375
+            = 0.148781015625
+        total_profit_percentage = (open_value/close_value) - 1
+            = (0.29925/0.150468984375)-1
+            = 0.9887819489377738
+    """
+    trade = Trade(
+        pair='ETH/BTC',
+        stake_amount=0.001,
+        open_rate=0.02,
+        amount=5,
+        is_open=True,
+        fee_open=fee.return_value,
+        fee_close=fee.return_value,
+        open_date=five_hours_ago,
+        exchange='kraken',
+        is_short=True,
+        leverage=3.0,
+        interest_rate=0.0005
+    )
+    assert trade.close_profit is None
+    assert trade.close_date is None
+    assert trade.is_open is True
+    trade.close(0.01)
+    assert trade.is_open is False
+    assert trade.close_profit == 0.98878195
+    assert trade.close_date is not None
+
+    # TODO-mg: Remove these comments probably
     #new_date = arrow.Arrow(2020, 2, 2, 15, 6, 1).datetime,
     # assert trade.close_date != new_date
     # # Close should NOT update close_date if the trade has been closed already