import glob
import pytest
from PIL import Image
from utils import init_files_setup, run_cmd, ctx_run_cmd, OUTPUTS
# ------- Files setup ------------------------------------------------------------------------------------------------ #
# Retrieve INPUTS
INPUTS = init_files_setup()
CMD_PREFIX = ['platform', 'model', 'infer']
def run_infer(*args, **kwargs):
run_cmd(CMD_PREFIX, *args, **kwargs)
# ------- Image Input Tests ------------------------------------------------------------------------------------------ #
@pytest.mark.parametrize('outputs,expected',
[([OUTPUTS['IMAGE']], {
'expect_nb_image': 1
}), ([OUTPUTS['VIDEO']], {
'expect_nb_video': 1
}), ([OUTPUTS['STD']], {}),
([OUTPUTS['INT_WILDCARD_JSON']], {
'expect_nb_json': 1
}),
([OUTPUTS['STR_WILDCARD_JSON']], {
'expect_nb_json': 1
}),
# coding: utf-8
from utils import init_files_setup
if __name__ == '__main__':
init_files_setup()