popov/neuron-tai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
Dobromir Popov
2026-07-11 22:47:12 +03:00
parent 7cf8d9bcf3
commit f615b6befb
3 changed files with 30 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/test_billing_ledger.py
View File

@@ -511,10 +511,12 @@ def test_proxy_chat_splits_payout_by_tracker_assigned_route_span():
tracker_url = f"http://127.0.0.1:{port}"
head = _UsageStubNode(total_tokens=1000)
head_port = head.start()
tail = _UsageStubNode(total_tokens=1000)
tail_port = tail.start()
try:
for endpoint, wallet, vram, bench in (
(f"http://127.0.0.1:{head_port}", "wallet-head", 5_000, 2.0),
("http://127.0.0.1:1", "wallet-tail", 10_000, 1.0),
for endpoint, wallet, vram, bench, shard_start, shard_end in (
(f"http://127.0.0.1:{head_port}", "wallet-head", 5_000, 2.0, 0, 3),
(f"http://127.0.0.1:{tail_port}", "wallet-tail", 10_000, 1.0, 4, 11),
):
reg = json.dumps({
"endpoint": endpoint,
@@ -528,8 +530,8 @@ def test_proxy_chat_splits_payout_by_tracker_assigned_route_span():
"tracker_mode": endpoint.endswith(str(head_port)),
"wallet_address": wallet,
"managed_assignment": True,
"shard_start": 0,
"shard_end": 999,
"shard_start": shard_start,
"shard_end": shard_end,
}).encode()
req = urllib.request.Request(
f"{tracker_url}/v1/nodes/register",
@@ -547,6 +549,7 @@ def test_proxy_chat_splits_payout_by_tracker_assigned_route_span():
assert ledger.get_node_pending("wallet-tail") == pytest.approx(pool * 8 / 12)
finally:
head.stop()
tail.stop()
tracker.stop()
@@ -628,7 +631,7 @@ def test_proxy_chat_records_validation_event_with_plain_route_metadata():
class FakeRegistry:
def get_wallet(self, wallet_address):
return type("Wallet", (), {"banned": False})()
return type("Wallet", (), {"banned": False, "reputation": 1.0})()
class FakeValidation:
def __init__(self):