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