Skip to content

Commit f20d7d6

Browse files
authored
Feat/rao changes (#10)
* update lib and pyi for NeuronInfo fields * bump ver * change ver to pre-release * chore: ruff * chore: ruff * use tuple instead * use older version of bittensor for testing * add more logging to test and attr fix * use new neuron info for tests * copy over test utils; stop using bittensor for comp * use test utils for all tests
1 parent 65e9d2d commit f20d7d6

File tree

8 files changed

+1264
-20
lines changed

8 files changed

+1264
-20
lines changed

tests/neurons.hex

Lines changed: 1 addition & 1 deletion
Large diffs are not rendered by default.

tests/neurons_lite.hex

Lines changed: 1 addition & 1 deletion
Large diffs are not rendered by default.

tests/test_decode_delegate_info.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
fix_field as fix_field_fixes,
1212
py_getattr as py_getattr_fixes,
1313
)
14+
from .utils import chain_data
1415

1516
TEST_DELEGATE_INFO_HEX = {
1617
"delegated normal": lambda: get_file_bytes("tests/delegated_info.hex"),
@@ -54,7 +55,7 @@ def test_decode_delegated_matches_python_impl(self):
5455
)
5556
)
5657

57-
delegate_info_py_list = bittensor.DelegateInfo.delegated_list_from_vec_u8(
58+
delegate_info_py_list = chain_data.DelegateInfo.delegated_list_from_vec_u8(
5859
list(TEST_DELEGATE_INFO_HEX["delegated normal"]())
5960
)
6061

@@ -102,8 +103,8 @@ def test_decode_vec_matches_python_impl(self):
102103
bt_decode.DelegateInfo.decode_vec(TEST_DELEGATE_INFO_HEX["vec normal"]())
103104
)
104105

105-
delegates_info_py: List[bittensor.DelegateInfo] = (
106-
bittensor.DelegateInfo.list_from_vec_u8(
106+
delegates_info_py: List[chain_data.DelegateInfo] = (
107+
chain_data.DelegateInfo.list_from_vec_u8(
107108
list(TEST_DELEGATE_INFO_HEX["vec normal"]())
108109
)
109110
)

tests/test_decode_neurons.py

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -4,22 +4,24 @@
44
import netaddr
55
import unittest
66

7-
import bt_decode
87
import bittensor
8+
import bt_decode
99

1010
from . import (
1111
get_file_bytes,
1212
fix_field as fix_field_fixes,
1313
py_getattr as py_getattr_fixes,
1414
)
1515

16+
from .utils import chain_data
17+
1618
TEST_NEURON_INFO_LITE_HEX = {
17-
"normal": "c40352ca71e26e83b6c86058fd4d3c9643ea5dc11f120a7c80f47ec5770b457d8853018ca894cb3d02aaf9b96741c831a3970cf250a58ec46e6a66f269be0b4b040400ba94330000000000c7020000e0aaf22c000000000000000000000000ad240404000000000000000000000000000000000000000000000000000000000000000000048853018ca894cb3d02aaf9b96741c831a3970cf250a58ec46e6a66f269be0b4b1ee07d572901f6fc6f002901210166e7030000006e1e9b00007d05",
19+
"normal": "fe65717dad0447d715f660a0a58411de509b42e6efb8375f562f58a554d5860e1cbd2d43530a44705ad088af313e18f80b53ef16b36177cd4b77b846f2a5f07c0008010000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000041cbd2d43530a44705ad088af313e18f80b53ef16b36177cd4b77b846f2a5f07c0bbb22f41921010bbb22f41921010bbb22f4192101000007989cc65f0100000000009801feff0300",
1820
"vec normal": lambda: get_file_bytes("tests/neurons_lite.hex"),
1921
}
2022

2123
TEST_NEURON_INFO_HEX = {
22-
"normal": "c40352ca71e26e83b6c86058fd4d3c9643ea5dc11f120a7c80f47ec5770b457d8853018ca894cb3d02aaf9b96741c831a3970cf250a58ec46e6a66f269be0b4b040400ba94330000000000c7020000e0aaf22c000000000000000000000000ad240404000000000000000000000000000000000000000000000000000000000000000000048853018ca894cb3d02aaf9b96741c831a3970cf250a58ec46e6a66f269be0b4b6220f458c056ce4900c0bc4276030000006e1e9b00000404feff0300009d03",
24+
"normal": "d43593c715fdd31c61141abd04a99fd6822c8558854ccde39a5684e7a56da27d1cbd2d43530a44705ad088af313e18f80b53ef16b36177cd4b77b846f2a5f07c0408010000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000041cbd2d43530a44705ad088af313e18f80b53ef16b36177cd4b77b846f2a5f07c00000000000000000000002d0101000000",
2325
"vec normal": lambda: get_file_bytes("tests/neurons.hex"),
2426
}
2527

@@ -69,7 +71,7 @@ def test_decode_matches_python_impl(self):
6971
bytes.fromhex(TEST_NEURON_INFO_LITE_HEX["normal"])
7072
)
7173

72-
neuron_info_py = bittensor.NeuronInfoLite.from_vec_u8(
74+
neuron_info_py = chain_data.NeuronInfoLite.from_vec_u8(
7375
list(bytes.fromhex(TEST_NEURON_INFO_LITE_HEX["normal"]))
7476
)
7577

@@ -112,8 +114,8 @@ def test_decode_vec_matches_python_impl(self):
112114
)
113115
)
114116

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(
117119
list(TEST_NEURON_INFO_LITE_HEX["vec normal"]())
118120
)
119121
)
@@ -159,7 +161,7 @@ def test_decode_matches_python_impl(self):
159161
bytes.fromhex(TEST_NEURON_INFO_HEX["normal"])
160162
)
161163

162-
neuron_info_py = bittensor.NeuronInfo.from_vec_u8(
164+
neuron_info_py = chain_data.NeuronInfo.from_vec_u8(
163165
list(bytes.fromhex(TEST_NEURON_INFO_HEX["normal"]))
164166
)
165167

@@ -198,8 +200,8 @@ def test_decode_vec_matches_python_impl(self):
198200
TEST_NEURON_INFO_HEX["vec normal"]()
199201
)
200202

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(
203205
list(TEST_NEURON_INFO_HEX["vec normal"]())
204206
)
205207
)

tests/test_decode_stake_info.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
fix_field as fix_field_fixes,
1111
py_getattr as py_getattr_fixes,
1212
)
13+
from .utils import chain_data
1314

1415
TEST_STAKE_INFO_HEX = {
1516
"vec normal": "08e4df2c7397e1443378b4cec0f2fca9dac1d0923d020e7aab11dd41428014ab595c40bc195cb2fd36b8b0e2397087c73b555b81e0bfe2975a40b9f78e039d44420759e02b6017ae4f8eac06ab73ff50aa97c0aafd27cd5c311e2fbbe5628f24901f4e3e1b06695c40bc195cb2fd36b8b0e2397087c73b555b81e0bfe2975a40b9f78e039d4442e25c4a0c",
@@ -49,8 +50,8 @@ def test_decode_vec_matches_python_impl(self):
4950
bytes.fromhex(TEST_STAKE_INFO_HEX["vec normal"])
5051
)
5152

52-
stake_info_py_list: List[bittensor.StakeInfo] = (
53-
bittensor.StakeInfo.list_from_vec_u8(
53+
stake_info_py_list: List[chain_data.StakeInfo] = (
54+
chain_data.StakeInfo.list_from_vec_u8(
5455
list(bytes.fromhex(TEST_STAKE_INFO_HEX["vec normal"]))
5556
)
5657
)
@@ -95,8 +96,8 @@ def test_decode_vec_vec_matches_python_impl(self):
9596
)
9697

9798
# Poor method name, should be dict_of_list_from_vec_u8
98-
stake_info_py_dict: Dict[str, List[bittensor.StakeInfo]] = (
99-
bittensor.StakeInfo.list_of_tuple_from_vec_u8(
99+
stake_info_py_dict: Dict[str, List[chain_data.StakeInfo]] = (
100+
chain_data.StakeInfo.list_of_tuple_from_vec_u8(
100101
list(bytes.fromhex(TEST_STAKE_INFO_HEX["vec vec normal"]))
101102
)
102103
)

tests/test_decode_subnet_info.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
fix_field as fix_field_fixes,
1212
py_getattr as py_getattr_fixes,
1313
)
14+
from .utils import chain_data
1415

1516
TEST_SUBNET_INFO_HEX = {
1617
"normal": "0828feff010013ffffffffffffffff214e010104feff0300c8010401040d03a1050000c28ff4070398b6d54370c07a546ab0bab5ca9847eb5890ada1bda127633e607097ad4517dd2ca0f010",
@@ -56,7 +57,7 @@ def test_decode_matches_python_impl(self):
5657
bytes.fromhex(TEST_SUBNET_INFO_HEX["normal"])
5758
)
5859

59-
subnet_info_py = bittensor.SubnetInfo.from_vec_u8(
60+
subnet_info_py = chain_data.SubnetInfo.from_vec_u8(
6061
list(bytes.fromhex(TEST_SUBNET_INFO_HEX["normal"]))
6162
)
6263

@@ -120,7 +121,7 @@ def test_decode_vec_option_matches_python_impl(self):
120121
)
121122

122123
subnet_info_list_py = (
123-
bittensor.SubnetInfo.list_from_vec_u8( # Option specified internally
124+
chain_data.SubnetInfo.list_from_vec_u8( # Option specified internally
124125
TEST_SUBNET_INFO_HEX["vec option normal"]()
125126
)
126127
)

tests/utils/__init__.py

Whitespace-only changes.

0 commit comments

Comments
 (0)