|
4 | 4 | import netaddr
|
5 | 5 | import unittest
|
6 | 6 |
|
7 |
| -import bt_decode |
8 | 7 | import bittensor
|
| 8 | +import bt_decode |
9 | 9 |
|
10 | 10 | from . import (
|
11 | 11 | get_file_bytes,
|
12 | 12 | fix_field as fix_field_fixes,
|
13 | 13 | py_getattr as py_getattr_fixes,
|
14 | 14 | )
|
15 | 15 |
|
| 16 | +from .utils import chain_data |
| 17 | + |
16 | 18 | TEST_NEURON_INFO_LITE_HEX = {
|
17 |
| - "normal": "c40352ca71e26e83b6c86058fd4d3c9643ea5dc11f120a7c80f47ec5770b457d8853018ca894cb3d02aaf9b96741c831a3970cf250a58ec46e6a66f269be0b4b040400ba94330000000000c7020000e0aaf22c000000000000000000000000ad240404000000000000000000000000000000000000000000000000000000000000000000048853018ca894cb3d02aaf9b96741c831a3970cf250a58ec46e6a66f269be0b4b1ee07d572901f6fc6f002901210166e7030000006e1e9b00007d05", |
| 19 | + "normal": "fe65717dad0447d715f660a0a58411de509b42e6efb8375f562f58a554d5860e1cbd2d43530a44705ad088af313e18f80b53ef16b36177cd4b77b846f2a5f07c0008010000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000041cbd2d43530a44705ad088af313e18f80b53ef16b36177cd4b77b846f2a5f07c0bbb22f41921010bbb22f41921010bbb22f4192101000007989cc65f0100000000009801feff0300", |
18 | 20 | "vec normal": lambda: get_file_bytes("tests/neurons_lite.hex"),
|
19 | 21 | }
|
20 | 22 |
|
21 | 23 | TEST_NEURON_INFO_HEX = {
|
22 |
| - "normal": "c40352ca71e26e83b6c86058fd4d3c9643ea5dc11f120a7c80f47ec5770b457d8853018ca894cb3d02aaf9b96741c831a3970cf250a58ec46e6a66f269be0b4b040400ba94330000000000c7020000e0aaf22c000000000000000000000000ad240404000000000000000000000000000000000000000000000000000000000000000000048853018ca894cb3d02aaf9b96741c831a3970cf250a58ec46e6a66f269be0b4b6220f458c056ce4900c0bc4276030000006e1e9b00000404feff0300009d03", |
| 24 | + "normal": "d43593c715fdd31c61141abd04a99fd6822c8558854ccde39a5684e7a56da27d1cbd2d43530a44705ad088af313e18f80b53ef16b36177cd4b77b846f2a5f07c0408010000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000041cbd2d43530a44705ad088af313e18f80b53ef16b36177cd4b77b846f2a5f07c00000000000000000000002d0101000000", |
23 | 25 | "vec normal": lambda: get_file_bytes("tests/neurons.hex"),
|
24 | 26 | }
|
25 | 27 |
|
@@ -69,7 +71,7 @@ def test_decode_matches_python_impl(self):
|
69 | 71 | bytes.fromhex(TEST_NEURON_INFO_LITE_HEX["normal"])
|
70 | 72 | )
|
71 | 73 |
|
72 |
| - neuron_info_py = bittensor.NeuronInfoLite.from_vec_u8( |
| 74 | + neuron_info_py = chain_data.NeuronInfoLite.from_vec_u8( |
73 | 75 | list(bytes.fromhex(TEST_NEURON_INFO_LITE_HEX["normal"]))
|
74 | 76 | )
|
75 | 77 |
|
@@ -112,8 +114,8 @@ def test_decode_vec_matches_python_impl(self):
|
112 | 114 | )
|
113 | 115 | )
|
114 | 116 |
|
115 |
| - neurons_info_py: List[bittensor.NeuronInfoLite] = ( |
116 |
| - bittensor.NeuronInfoLite.list_from_vec_u8( |
| 117 | + neurons_info_py: List[chain_data.NeuronInfoLite] = ( |
| 118 | + chain_data.NeuronInfoLite.list_from_vec_u8( |
117 | 119 | list(TEST_NEURON_INFO_LITE_HEX["vec normal"]())
|
118 | 120 | )
|
119 | 121 | )
|
@@ -159,7 +161,7 @@ def test_decode_matches_python_impl(self):
|
159 | 161 | bytes.fromhex(TEST_NEURON_INFO_HEX["normal"])
|
160 | 162 | )
|
161 | 163 |
|
162 |
| - neuron_info_py = bittensor.NeuronInfo.from_vec_u8( |
| 164 | + neuron_info_py = chain_data.NeuronInfo.from_vec_u8( |
163 | 165 | list(bytes.fromhex(TEST_NEURON_INFO_HEX["normal"]))
|
164 | 166 | )
|
165 | 167 |
|
@@ -198,8 +200,8 @@ def test_decode_vec_matches_python_impl(self):
|
198 | 200 | TEST_NEURON_INFO_HEX["vec normal"]()
|
199 | 201 | )
|
200 | 202 |
|
201 |
| - neurons_info_py: List[bittensor.NeuronInfo] = ( |
202 |
| - bittensor.NeuronInfo.list_from_vec_u8( |
| 203 | + neurons_info_py: List[chain_data.NeuronInfo] = ( |
| 204 | + chain_data.NeuronInfo.list_from_vec_u8( |
203 | 205 | list(TEST_NEURON_INFO_HEX["vec normal"]())
|
204 | 206 | )
|
205 | 207 | )
|
|
0 commit comments