code: def setUp(self):
      super().setUp()
      self._features = np.array([[1.0, 2.0], [3.0, 2.0]])
      self._num_actions = 3
      self._actions = np.array([1, 2])
      self._rewards = np.array([1.0, 1])
  
file_path: rlax/rlax/_src/embedding_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  import numpy
  from rlax._src import embedding
label_desc: Use image node_id 1 for calling the EmbeddingTest obj's underlying member method code with example usage: obj.setUp() without return types