Ejemplo n.º 1
0
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
Ejemplo n.º 2
0
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