def test_payload_load_json_array(): payload = Payload() payload.add_file(Path(__file__).parent.joinpath("data", "array_01.json")) json_data = payload.get_data() assert json_data assert json_data[1] == "value_1"
def test_payload_load_json_dictionary(): payload = Payload() payload.add_file( Path(__file__).parent.joinpath("data", "dictionary_01.json")) json_data = payload.get_data() assert json_data assert json_data["key"] == "value"
def test_payload_load_directory(): payload = Payload() payload.add_directory(Path(__file__).parent.joinpath("data")) objects = payload.all() assert len(objects) == 5 assert objects[0].type == "array" assert objects[0].name == "array_01.json" assert objects[0].extension == ".json"
def test_job_execute(): context = Context(mode=MODE.TEST, config_path=Path(__file__).parent.parent.joinpath( "data", "config")) job = Job(process=Process(mode=MODE.TEST, context=context)) output = job.execute(Payload()) assert output
def test_pipe_execute_parallel(): context = Context(mode=MODE.TEST, config_path=Path(__file__).parent.parent.joinpath( "data", "config")) job1 = Job(process=Process(mode=MODE.TEST, context=context)) job2 = Job(process=Process(mode=MODE.TEST, context=context)) pipe = Pipe(units=[job1, job2], execute_type=Pipe.Type.PARALLEL) output = pipe.execute(Payload()) assert output
def test_payload_load_named_image(): payload = Payload() payload.add_file(Path(__file__).parent.joinpath("data", "image_01.png"), name="test") image = payload.get_image() assert image image = payload.get_image("test") assert image image = payload.get_image("another") assert not image
def test_payload_create(): payload = Payload() assert payload
def test_payload_load_pdf(): payload = Payload() payload.add_file(Path(__file__).parent.joinpath("data", "pdf_01.pdf")) pdf_data = payload.get(object_type="pdf") assert pdf_data
def test_payload_load_image(): payload = Payload() payload.add_file(Path(__file__).parent.joinpath("data", "image_01.png")) image = payload.get_image() assert image