Function test_parallel_map_internal_client

test/espnet3/parallel/test_parallel.py:155–172

(local_cfg, monkeypatch)

@pytest.mark.execution_timeout(30)
def test_parallel_map_internal_client(local_cfg, monkeypatch):
    # Test the branch where client=None and get_client() is internally used
    class _Ctx:
        def __init__(self, client):
            self.client = client

        def __enter__(self):
            return self.client

        def __exit__(self, exc_type, exc, tb):
            self.client.close()

    cli = build_client(local_cfg)
    monkeypatch.setattr(
        "espnet3.parallel.parallel.get_client", lambda *a, **k: _Ctx(cli)
    )
    res = parallel_map(lambda x: x * 2, [1, 2, 3])
    assert res == [2, 4, 6]


# ------------------------------------------------------------