fix e2e test

Author: Roman

tests/e2e_tests/test_metagraph.py

@@ -206,22 +206,22 @@ def test_metagraph_info(subtensor, alice_wallet, bob_wallet):
         symbol="α",
         identity=None,
         network_registered_at=0,
-        owner_hotkey=NULL_KEY,
-        owner_coldkey=NULL_KEY,
+        owner_hotkey="5C4hrfjw9DjXZTzV3MwzrrAr9P1MJhSrvWGWqi1eSuyUpnhM",
+        owner_coldkey="5C4hrfjw9DjXZTzV3MwzrrAr9P1MJhSrvWGWqi1eSuyUpnhM",
         block=1,
         tempo=100,
         last_step=0,
         blocks_since_last_step=1,
         subnet_emission=Balance(0),
-        alpha_in=Balance.from_tao(10),
-        alpha_out=Balance.from_tao(1),
+        alpha_in=Balance.from_tao(10).set_unit(netuid=alice_subnet_netuid),
+        alpha_out=Balance.from_tao(1).set_unit(netuid=alice_subnet_netuid),
         tao_in=Balance.from_tao(10),
-        alpha_out_emission=Balance(0),
-        alpha_in_emission=Balance(0),
+        alpha_out_emission=Balance(0).set_unit(netuid=alice_subnet_netuid),
+        alpha_in_emission=Balance(0).set_unit(netuid=alice_subnet_netuid),
         tao_in_emission=Balance(0),
-        pending_alpha_emission=Balance(0),
+        pending_alpha_emission=Balance(0).set_unit(netuid=alice_subnet_netuid),
         pending_root_emission=Balance(0),
-        subnet_volume=Balance(0),
+        subnet_volume=Balance(0).set_unit(netuid=alice_subnet_netuid),
         moving_price=Balance(0),
         rho=10,
         kappa=32767,
@@ -272,16 +272,16 @@ def test_metagraph_info(subtensor, alice_wallet, bob_wallet):
         validator_permit=(False,),
         pruning_score=[0.0],
         last_update=(0,),
-        emission=[Balance(0)],
+        emission=[Balance(0).set_unit(alice_subnet_netuid)],
         dividends=[0.0],
         incentives=[0.0],
         consensus=[0.0],
         trust=[0.0],
         rank=[0.0],
         block_at_registration=(0,),
-        alpha_stake=[Balance.from_tao(1.0)],
+        alpha_stake=[Balance.from_tao(1.0).set_unit(alice_subnet_netuid)],
         tao_stake=[Balance(0)],
-        total_stake=[Balance.from_tao(1.0)],
+        total_stake=[Balance.from_tao(1.0).set_unit(alice_subnet_netuid)],
         tao_dividends_per_hotkey=[
             ("5C4hrfjw9DjXZTzV3MwzrrAr9P1MJhSrvWGWqi1eSuyUpnhM", Balance(0))
         ],
@@ -299,18 +299,18 @@ def test_metagraph_info(subtensor, alice_wallet, bob_wallet):
             symbol="Τ",
             identity=None,
             network_registered_at=0,
-            owner_hotkey=NULL_KEY,
-            owner_coldkey=NULL_KEY,
+            owner_hotkey="5C4hrfjw9DjXZTzV3MwzrrAr9P1MJhSrvWGWqi1eSuyUpnhM",
+            owner_coldkey="5C4hrfjw9DjXZTzV3MwzrrAr9P1MJhSrvWGWqi1eSuyUpnhM",
             block=1,
             tempo=100,
             last_step=0,
             blocks_since_last_step=1,
             subnet_emission=Balance(0),
-            alpha_in=Balance(0),
-            alpha_out=Balance(0),
+            alpha_in=Balance(0).set_unit(netuid=alice_subnet_netuid),
+            alpha_out=Balance(0).set_unit(netuid=alice_subnet_netuid),
             tao_in=Balance(0),
-            alpha_out_emission=Balance(0),
-            alpha_in_emission=Balance(0),
+            alpha_out_emission=Balance(0).set_unit(netuid=alice_subnet_netuid),
+            alpha_in_emission=Balance(0).set_unit(netuid=alice_subnet_netuid),
             tao_in_emission=Balance(0),
             pending_alpha_emission=Balance(0),
             pending_root_emission=Balance(0),
@@ -420,8 +420,8 @@ def test_metagraph_info(subtensor, alice_wallet, bob_wallet):
         netuid=alice_subnet_netuid, block=block
     )
 
-    assert metagraph_info.owner_coldkey == (tuple(alice_wallet.hotkey.public_key),)
-    assert metagraph_info.owner_hotkey == (tuple(alice_wallet.coldkey.public_key),)
+    assert metagraph_info.owner_coldkey == alice_wallet.hotkey.ss58_address
+    assert metagraph_info.owner_hotkey == alice_wallet.coldkey.ss58_address
 
     metagraph_infos = subtensor.get_all_metagraphs_info(block)