【Python】一行で値を確認する

コマンドライン引数の値を確認したいなっと思っていて， assertが思い出せなかったのでメモする．

assert文の使い方は，以下の通り

assert [条件式], [Falseの場合に出力するメッセージ]

今回はこんな感じで，コマンドラインの引数確認に使いました．

assert args.mode is 'sim' or args.mode is 'real', 'ERROR'

ま〜よく考えると，argparseモジュールのchoicesオプションをつけたほうがスマートですが

parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--mode',
    type=str,
    choices=['sim', 'real'],
    required=True,
    help='select mode, [sim] or [real]')