code: def test_basic(func):
      x = da.random.default_rng().random((2, 3, 4), chunks=(1, 2, 2))
      x[x < 0.4] = 0
  
      y = da.ma.masked_equal(x, 0)
  
      xx = func(x)
      yy = func(y)
  
      assert_eq(xx, da.ma.filled(yy, 0))
  
      if yy.shape:
          zz = yy.compute()
          assert isinstance(zz, np.ma.masked_array)
  
imports: from __future__ import annotations
  import random
  import sys
  from copy import deepcopy
  from itertools import product
  import numpy
  import pytest
  import dask.array
  from dask.array.numpy_compat import ComplexWarning, _numpy_123
  from dask.array.utils import assert_eq
  from dask.base import tokenize
  from dask.utils import typename
label_desc: Use image node_id 4 for calling a global function with example usage: test_basic(func) without return types