class_docstr: Tests methods specific to the OneHotCategorical distribution
class_docstr_tok: ["Tests","methods","specific","to","the","OneHotCategorical","distribution"]
code: def test_support_non_vectorized(self):
      s = dist.OneHotCategorical(
          self.d_ps[0].squeeze(0)
      ).enumerate_support()
      assert_equal(s.data, self.support_one_hot_non_vec)
  
file_path: pyro/tests/distributions/test_one_hot_categorical.py
imports: from unittest import TestCase
  import numpy
  import pytest
  import torch
  import pyro.distributions
  from tests.common import assert_equal
label_desc: Use image node_id 2 for calling the TestOneHotCategorical obj's underlying member method code with example usage: obj.test_support_non_vectorized() without return types