def METHOD_NAME(model_func, use_workspace_io, expected_memory_pressure):
    """
    Test that memory pressure is correctly calculated for NPU external functions.
    """
    mod = model_func()
    with tvm.transform.PassContext(config={"tir.usmp.use_workspace_io": use_workspace_io}):
        memory_pressure = _calculate_memory_pressure(mod)
    assert memory_pressure == expected_memory_pressure