code: def test_multinomial(generator_class):
      for size, chunks in [(5, 3), ((5, 4), (2, 3))]:
          x = generator_class().multinomial(
              20, [1 / 6.0] * 6, size=size, chunks=chunks
          )
          y = np.random.multinomial(20, [1 / 6.0] * 6, size=size)
  
          assert x.shape == y.shape == x.compute().shape
  
imports: from __future__ import annotations
  import pytest
  import numpy
  import dask
  import dask.array
  from dask.array.core import Array
  from dask.array.utils import assert_eq
  from dask.multiprocessing import _dumps, _loads
  from dask.utils import key_split
label_desc: Use image node_id 23 for calling a global function with example usage: test_multinomial(generator_class) without return types