code: def test_embed_zeros(self):
      # Embedding zero feature array.
      zero_features = np.zeros_like(self._features)
      emb = embedding.embed_oar(
          zero_features, self._actions, self._rewards, self._num_actions
      )
      np.testing.assert_array_equal(
          emb[:, : self._features.shape[-1]], zero_features
      )
  
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 2 for calling the EmbeddingTest obj's underlying member method code with example usage: obj.test_embed_zeros() without return types