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)")
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)")