enable solana single method tests in cli

chainstacklabs · Apr 25, 2024 · 8fa6ff2 · 8fa6ff2
1 parent 7387cc6
commit 8fa6ff2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/chainbench/user/methods/__init__.py b/chainbench/user/methods/__init__.py
@@ -5,10 +5,12 @@
 from .common import get_subclass_functions
 from .ethereum import EthBeaconMethods
 from .evm import EvmMethods
+from .solana import SolanaMethods
 
 __all__ = [
     "EthBeaconMethods",
     "EvmMethods",
+    "SolanaMethods",
     "get_subclass_functions",
     "all_method_classes",
     "all_methods",
@@ -18,6 +20,7 @@
 all_method_classes: list[Type[EthBeaconMethods | EvmMethods]] = [
     EthBeaconMethods,
     EvmMethods,
+    SolanaMethods,
 ]
 
 

diff --git a/chainbench/user/methods/solana.py b/chainbench/user/methods/solana.py
@@ -14,7 +14,7 @@ class SolanaMethods(SolanaUser):
     def task_to_method(task_name: str) -> str:
         task_name_stripped = task_name.replace("_task", "")
         words = task_name_stripped.split("_")
-        method = "".join([words[1]] + [word.capitalize() for word in words[2:]])
+        method = "".join([words[0]] + [word.capitalize() for word in words[1:]])
         return method
 
     def method_to_task_function(self, method: str) -> t.Callable: