def test_wf_container_task():

    @task
    def t1(a: int) -> (int, str):
        return ((a + 2), (str(a) + '-HELLO'))
    t2 = ContainerTask('raw', image='alpine', inputs=kwtypes(a=int, b=str), input_data_dir='/tmp', output_data_dir='/tmp', command=['cat'], arguments=['/tmp/a'])

    def wf(a: int):
        (x, y) = t1(a=a)
        t2(a=x, b=y)
    with task_mock(t2) as mock:
        mock.side_effect = (lambda a, b: None)
        assert (t2(a=10, b='hello') is None)
        wf(a=10)
def test_wf_container_task():

    @task
    def t1(a: int) -> (int, str):
        return ((a + 2), (str(a) + '-HELLO'))
    t2 = ContainerTask('raw', image='alpine', inputs=kwtypes(a=int, b=str), input_data_dir='/tmp', output_data_dir='/tmp', command=['cat'], arguments=['/tmp/a'])

    def wf(a: int):
        (x, y) = t1(a=a)
        t2(a=x, b=y)
    with task_mock(t2) as mock:
        mock.side_effect = (lambda a, b: None)
        assert (t2(a=10, b='hello') is None)
        wf(a=10)