Skip to content

Commit 1d38d08

Browse files
committed
enable solana single method tests in cli
1 parent bd8ec5a commit 1d38d08

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

chainbench/user/methods/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,12 @@
55
from .common import get_subclass_functions
66
from .ethereum import EthBeaconMethods
77
from .evm import EvmMethods
8+
from .solana import SolanaMethods
89

910
__all__ = [
1011
"EthBeaconMethods",
1112
"EvmMethods",
13+
"SolanaMethods",
1214
"get_subclass_functions",
1315
"all_method_classes",
1416
"all_methods",
@@ -18,6 +20,7 @@
1820
all_method_classes: list[Type[EthBeaconMethods | EvmMethods]] = [
1921
EthBeaconMethods,
2022
EvmMethods,
23+
SolanaMethods,
2124
]
2225

2326

chainbench/user/methods/solana.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ class SolanaMethods(SolanaUser):
1414
def task_to_method(task_name: str) -> str:
1515
task_name_stripped = task_name.replace("_task", "")
1616
words = task_name_stripped.split("_")
17-
method = "".join([words[1]] + [word.capitalize() for word in words[2:]])
17+
method = "".join([words[0]] + [word.capitalize() for word in words[1:]])
1818
return method
1919

2020
def method_to_task_function(self, method: str) -> t.Callable:

0 commit comments

Comments
 (0)