dekun
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
"""粉尘阈值 / 卖回封顶 / 买入增量逻辑相关单测."""
|
|
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
from lib.options.options_spot_bridge_lib import (
|
|
display_sellable_coin,
|
|
is_sellable_spot_coin,
|
|
spot_min_sell_sz,
|
|
)
|
|
|
|
|
|
class TestSpotDust(unittest.TestCase):
|
|
def test_min_sz_fallback(self):
|
|
self.assertAlmostEqual(spot_min_sell_sz(None, "ETH"), 0.0001)
|
|
self.assertAlmostEqual(spot_min_sell_sz(None, "BTC"), 0.00001)
|
|
|
|
def test_dust_not_sellable(self):
|
|
self.assertFalse(is_sellable_spot_coin(None, "ETH", 0.000004))
|
|
self.assertFalse(is_sellable_spot_coin(None, "BTC", 0.000002))
|
|
self.assertEqual(display_sellable_coin(None, "ETH", 0.000004), 0.0)
|
|
|
|
def test_above_min_sellable(self):
|
|
self.assertTrue(is_sellable_spot_coin(None, "ETH", 0.0001))
|
|
self.assertTrue(is_sellable_spot_coin(None, "BTC", 0.00001))
|
|
self.assertGreater(display_sellable_coin(None, "ETH", 0.001), 0)
|
|
|
|
|
|
class TestSellCapLogic(unittest.TestCase):
|
|
def test_excess_cap_math(self):
|
|
bought, keep = 0.01, 0.004
|
|
excess = max(0.0, bought - keep)
|
|
self.assertAlmostEqual(excess, 0.006)
|
|
sell_amt = min(0.02, excess)
|
|
self.assertAlmostEqual(sell_amt, 0.006)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|