code: def test_too_large_allocation():
      allocator = BlockedAllocator(16)
  
      with pytest.raises(ValueError):
          allocator.allocate(17)
  
file_path: DeepSpeed/tests/unit/inference/v2/ragged/test_blocked_allocator.py
imports: import random
  from typing import List
  import pytest
  import torch
  from deepspeed.inference.v2.ragged.blocked_allocator import BlockedAllocator
label_desc: Use image node_id 3 for calling a global function with example usage: test_too_large_allocation() without return types