code: def test_max_size():
      param = torch.rand((20, 30))
  
      bucket = ParamBucket(5, param.dtype, param.device)
      with pytest.raises(AssertionError):
          bucket.add_param(param)
  
imports: import pytest
  import torch
  from fairscale.nn.misc import ParamBucket
label_desc: Use image node_id 2 for calling a global function with example usage: test_max_size() without return types