Daniel Sapkota
92 lines
3.1 KiB
Python
92 lines
3.1 KiB
Python
"""Tests for market parsing, scoring, and the SQLite catalog store."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
|
|
from polymaker.catalog.gamma import parse_market
|
|
from polymaker.catalog.scoring import score_market
|
|
from polymaker.catalog.store import CatalogStore
|
|
|
|
RAW = {
|
|
"conditionId": "0xabc",
|
|
"question": "Will candidate X win?",
|
|
"slug": "will-x-win",
|
|
"clobTokenIds": json.dumps(["tok-yes", "tok-no"]),
|
|
"outcomes": json.dumps(["Yes", "No"]),
|
|
"orderPriceMinTickSize": 0.01,
|
|
"orderMinSize": 5,
|
|
"negRisk": True,
|
|
"acceptingOrders": True,
|
|
"rewardsMinSize": 10,
|
|
"rewardsMaxSpread": 3.0,
|
|
"feesEnabled": True,
|
|
"feeSchedule": {"rate": 0.01, "takerOnly": True, "rebateRate": 0.25},
|
|
"bestBid": 0.48,
|
|
"bestAsk": 0.50,
|
|
"liquidityNum": 20000.0,
|
|
"volumeNum": 500000.0,
|
|
"endDate": "2028-11-07T00:00:00Z",
|
|
"events": [{"id": 999, "slug": "2028-election"}],
|
|
}
|
|
|
|
|
|
def test_parse_market_maps_fields():
|
|
m = parse_market(RAW, reward_rates={"0xabc": 42.0})
|
|
assert m is not None
|
|
assert m.condition_id == "0xabc"
|
|
assert m.yes.token_id == "tok-yes"
|
|
assert m.no.token_id == "tok-no"
|
|
assert m.tick_size == 0.01
|
|
assert m.neg_risk is True
|
|
assert m.rewards_daily_rate == 42.0
|
|
assert m.taker_fee_bps == 100 # 0.01 -> 100 bps
|
|
assert m.maker_fee_bps == 0 # V2 makers pay zero
|
|
assert m.rebate_rate == 0.25
|
|
assert m.event_id == "999"
|
|
|
|
|
|
def test_parse_market_rejects_non_binary_and_closed():
|
|
triple = {**RAW, "clobTokenIds": json.dumps(["a", "b", "c"]),
|
|
"outcomes": json.dumps(["A", "B", "C"])}
|
|
assert parse_market(triple) is None
|
|
not_accepting = {**RAW, "acceptingOrders": False}
|
|
assert parse_market(not_accepting) is None
|
|
|
|
|
|
def test_score_prefers_rewards_and_rebates():
|
|
good = parse_market(RAW, {"0xabc": 100.0})
|
|
poor = parse_market({**RAW, "conditionId": "0xdef", "rewardsMinSize": 0,
|
|
"rewardsMaxSpread": 0, "feesEnabled": False},
|
|
{"0xdef": 0.0})
|
|
assert score_market(good).score > score_market(poor).score
|
|
|
|
|
|
def test_score_penalizes_extremity():
|
|
balanced = parse_market(RAW, {"0xabc": 50.0})
|
|
extreme = parse_market({**RAW, "conditionId": "0xext", "bestBid": 0.96, "bestAsk": 0.98},
|
|
{"0xext": 50.0})
|
|
assert score_market(extreme).extremity > score_market(balanced).extremity
|
|
|
|
|
|
def test_store_roundtrip_and_top(tmp_path):
|
|
store = CatalogStore(tmp_path / "s.db")
|
|
m = parse_market(RAW, {"0xabc": 42.0})
|
|
store.upsert_market(m)
|
|
assert store.get("0xabc").condition_id == "0xabc"
|
|
assert store.get_by_slug("will-x-win").slug == "will-x-win"
|
|
top = store.top(10)
|
|
assert len(top) == 1 and top[0][0].condition_id == "0xabc"
|
|
# tokens survive the JSON round-trip as a 2-tuple
|
|
assert len(store.get("0xabc").tokens) == 2
|
|
store.close()
|
|
|
|
|
|
def test_store_upsert_is_idempotent(tmp_path):
|
|
store = CatalogStore(tmp_path / "s.db")
|
|
m = parse_market(RAW, {"0xabc": 42.0})
|
|
store.upsert_market(m)
|
|
store.upsert_market(m) # second time updates, not duplicates
|
|
assert len(store.top(10)) == 1
|
|
store.close()
|