dekun
41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
"""OKX 公开 K 线小测(无网络):instId / bar 映射."""
|
|
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
|
|
class OkxPublicOhlcvMappingTests(unittest.TestCase):
|
|
def test_swap_inst_and_bar(self):
|
|
# 避免 import 整站 app:直接复刻映射逻辑做契约校验
|
|
def swap_inst(exchange_symbol: str) -> str:
|
|
sym = str(exchange_symbol or "").strip().upper()
|
|
if not sym:
|
|
return ""
|
|
if "/" in sym:
|
|
base, rest = sym.split("/", 1)
|
|
quote = rest.split(":", 1)[0].strip() or "USDT"
|
|
return f"{base.strip()}-{quote}-SWAP"
|
|
if sym.endswith("-SWAP"):
|
|
return sym
|
|
return sym
|
|
|
|
def public_bar(timeframe: str) -> str:
|
|
low = str(timeframe or "").strip().lower()
|
|
if low.endswith("m") and low[:-1].isdigit():
|
|
return low
|
|
if low.endswith("h") and low[:-1].isdigit():
|
|
return f"{low[:-1]}H"
|
|
if low.endswith("d") and low[:-1].isdigit():
|
|
return f"{low[:-1]}D"
|
|
return low or "15m"
|
|
|
|
self.assertEqual(swap_inst("ETH/USDT:USDT"), "ETH-USDT-SWAP")
|
|
self.assertEqual(swap_inst("BTC/USDT:USDT"), "BTC-USDT-SWAP")
|
|
self.assertEqual(public_bar("15m"), "15m")
|
|
self.assertEqual(public_bar("1h"), "1H")
|
|
self.assertEqual(public_bar("4h"), "4H")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|