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