code: def test_OneHotRound(self) -> None:
      self.assertTrue(
          np.allclose(
              strict_onehot_round(np.array([0.1, 0.5, 0.3])),
              np.array([0, 1, 0]),
          )
      )
      # One item should be set to one at random.
      self.assertEqual(
          np.count_nonzero(
              np.isclose(
                  randomized_onehot_round(np.array([0.0, 0.0, 0.0])),
                  np.array([1, 1, 1]),
              )
          ),
          1,
      )
  
imports: import numpy
  from ax.modelbridge.transforms.rounding import randomized_onehot_round, strict_onehot_round
  from ax.utils.common.testutils import TestCase
label_desc: Use image node_id 2 for calling the RoundingTest obj's underlying member method code with example usage: obj.test_OneHotRound() without return types