Python FileInput.handle_cliの例

コード例 #1

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

def test_file_input_cli(capsys, bin_file):
    test_file_input = FileInput()

    test_args = ["--input", bin_file]
    test_file_input.handle_cli(test_args, predict)
    out, _ = capsys.readouterr()
    assert '\\x810\\x899' in out

コード例 #2

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

def test_file_input_cli_list(capsys, bin_files):
    test_file_input = FileInput()

    test_args = ["--input"] + glob.glob(bin_files)
    test_file_input.handle_cli(test_args, predict)
    out, _ = capsys.readouterr()
    lines = out.strip().split('\n')
    for line in lines[-10:]:
        assert '\\x810\\x899' in line