diff --git a/src/cli.py b/src/shamira/cli.py copy from src/cli.py copy to src/shamira/cli.py --- a/src/cli.py +++ b/src/shamira/cli.py @@ -3,7 +3,8 @@ import sys from argparse import ArgumentParser -from shamira import generate, reconstruct, SException +from .core import generate, reconstruct, SException +from .benchmark import build_subparsers as build_benchmark def run(): @@ -12,6 +13,7 @@ def run(): build_split_parser(subparsers.add_parser("split")) build_join_parser(subparsers.add_parser("join")) + build_benchmark(subparsers.add_parser("benchmark")) parser.set_defaults(func=lambda _: parser.error("missing command"))