code: def test_sample_mask():
      space = Discrete(4, start=2)
      assert 2 <= space.sample() < 6
      assert (
          space.sample(mask=np.array([0, 1, 0, 0], dtype=np.int8)) == 3
      )
      assert (
          space.sample(mask=np.array([0, 0, 0, 0], dtype=np.int8)) == 2
      )
      assert space.sample(
          mask=np.array([0, 1, 0, 1], dtype=np.int8)
      ) in [3, 5]
  
label_desc: Use image node_id 2 for calling a global function with example usage: test_sample_mask() without return types