Python ImageInput.handle_cliの例

コード例 #1

ファイル: test_image_handler.py プロジェクト: umihui/BentoML

def test_image_input_cli(capsys, img_file):
    test_image_input = ImageInput()

    test_args = ["--input", img_file]
    test_image_input.handle_cli(test_args, predict)
    out, _ = capsys.readouterr()
    assert out.strip().endswith("(10, 10, 3)")

コード例 #2

ファイル: test_image_handler.py プロジェクト: umihui/BentoML

def test_image_input_cli_list(capsys, img_files):
    test_image_input = ImageInput()

    test_args = ["--input"] + glob.glob(img_files)
    test_image_input.handle_cli(test_args, predict)
    out, _ = capsys.readouterr()
    lines = out.strip().split('\n')
    for line in lines[-10:]:
        assert line.strip().endswith("(10, 10, 3)")