Skip to content

Commit 6667aec

Browse files
author
Roman
committed
make test_delegates compatible with both modes
1 parent 370e0a0 commit 6667aec

File tree

1 file changed

+18
-2
lines changed

1 file changed

+18
-2
lines changed

tests/e2e_tests/test_delegate.py

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ def test_change_take(local_chain, subtensor, alice_wallet, bob_wallet):
171171

172172

173173
@pytest.mark.asyncio
174-
async def test_delegates(subtensor, alice_wallet, bob_wallet):
174+
async def test_delegates(local_chain, subtensor, alice_wallet, bob_wallet):
175175
"""
176176
Tests:
177177
- Check default Delegates
@@ -240,6 +240,7 @@ async def test_delegates(subtensor, alice_wallet, bob_wallet):
240240
assert subtensor.get_delegated(bob_wallet.coldkey.ss58_address) == []
241241

242242
alice_subnet_netuid = subtensor.get_total_subnets() # 2
243+
set_tempo = 10
243244
# Register a subnet, netuid 2
244245
assert subtensor.register_subnet(alice_wallet), "Subnet wasn't created"
245246

@@ -250,6 +251,17 @@ async def test_delegates(subtensor, alice_wallet, bob_wallet):
250251

251252
assert wait_to_start_call(subtensor, alice_wallet, alice_subnet_netuid)
252253

254+
# set the same tempo for both type of nodes (fast and non-fast blocks)
255+
assert (
256+
sudo_set_admin_utils(
257+
local_chain,
258+
alice_wallet,
259+
call_function="sudo_set_tempo",
260+
call_params={"netuid": alice_subnet_netuid, "tempo": set_tempo},
261+
)[0]
262+
is True
263+
)
264+
253265
subtensor.add_stake(
254266
bob_wallet,
255267
alice_wallet.hotkey.ss58_address,
@@ -259,6 +271,9 @@ async def test_delegates(subtensor, alice_wallet, bob_wallet):
259271
wait_for_finalization=True,
260272
)
261273

274+
# let chain update validator_permits
275+
subtensor.wait_for_block(subtensor.block + set_tempo + 1)
276+
262277
bob_delegated = subtensor.get_delegated(bob_wallet.coldkey.ss58_address)
263278
assert bob_delegated == [
264279
DelegatedInfo(
@@ -272,9 +287,10 @@ async def test_delegates(subtensor, alice_wallet, bob_wallet):
272287
bob_delegated[0].total_daily_return.rao
273288
),
274289
netuid=alice_subnet_netuid,
275-
stake=get_dynamic_balance(bob_delegated[0].stake.rao),
290+
stake=get_dynamic_balance(bob_delegated[0].stake.rao, alice_subnet_netuid),
276291
),
277292
]
293+
bittensor.logging.console.success("Test [green]test_delegates[/green] passed.")
278294

279295

280296
def test_nominator_min_required_stake(local_chain, subtensor, alice_wallet, bob_wallet):

0 commit comments

Comments
 (0)