code: def test_pu_remove_noise():
      nm = np.array(
          [
              [0.9, 0.0, 0.0],
              [0.0, 0.7, 0.4],
              [0.1, 0.3, 0.6],
          ]
      )
      r = util.remove_noise_from_class(nm, 0)
      assert np.all(r - nm < 1e-4)
  
imports: from cleanlab.internal import util
  import numpy
  import pytest
  from cleanlab.internal.label_quality_utils import get_normalized_entropy
  from cleanlab.internal.multilabel_utils import int2onehot, onehot2int
  from cleanlab.internal.util import num_unique_classes, format_labels, get_missing_classes
  from cleanlab.internal.validation import assert_valid_class_labels
label_desc: Use image node_id 7 for calling a global function with example usage: test_pu_remove_noise() without return types