Python parse_argsの例

コード例 #1

def test_parse_args():
    mock_argv = ['clean_nerve']
    with mock.patch('sys.argv', mock_argv):
        args = clean_nerve.parse_args()

    assert not args.simulate
    assert args.cluster_type == 'infrastructure'

コード例 #2

def test_parse_args_cluster_type():
    mock_argv = ['clean_nerve', '--cluster-type', 'bar']
    with mock.patch('sys.argv', mock_argv):
        args = clean_nerve.parse_args()

    assert args.cluster_type == 'bar'

コード例 #3

def test_parse_args_simulate():
    mock_argv = ['clean_nerve', '--simulate']
    with mock.patch('sys.argv', mock_argv):
        args = clean_nerve.parse_args()

    assert args.simulate