Skip to content

Commit

Permalink
ruff
Browse files Browse the repository at this point in the history
  • Loading branch information
roman-opentensor committed Nov 12, 2024
1 parent 8fc6f6f commit aba9143
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions tests/unit_tests/test_async_subtensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -593,7 +593,9 @@ async def test_get_total_stake_for_coldkey(subtensor, mocker):
subtensor.substrate.query_multi = mocked_substrate_query_multi

# Call
result = await subtensor.get_total_stake_for_coldkey(*fake_addresses, block_hash=fake_block_hash)
result = await subtensor.get_total_stake_for_coldkey(
*fake_addresses, block_hash=fake_block_hash
)

assert mocked_substrate_create_storage_key.call_count == len(fake_addresses)
mocked_substrate_query_multi.assert_called_once()
Expand All @@ -613,7 +615,9 @@ async def test_get_total_stake_for_hotkey(subtensor, mocker):
subtensor.substrate.query_multiple = mocked_substrate_query_multiple

# Call
result = await subtensor.get_total_stake_for_hotkey(*fake_addresses, block_hash=fake_block_hash, reuse_block=reuse_block)
result = await subtensor.get_total_stake_for_hotkey(
*fake_addresses, block_hash=fake_block_hash, reuse_block=reuse_block
)

# Assertions
mocked_substrate_query_multiple.assert_called_once_with(
Expand Down

0 comments on commit aba9143

Please sign in to comment.