code: def test_extract_subsequences(self):
      output = model_learning.extract_subsequences(
          self.trajectories, self.start_indices, 3
      )
      expected_output = jnp.array(
          [
              [[0, 1, 4], [10, 20, 50]],
              [[1, 2, 5], [20, 30, 60]],
              [[2, 3, 6], [30, 40, 70]],
          ]
      )
      # Test output.
      np.testing.assert_allclose(output, expected_output)
  
file_path: rlax/rlax/_src/model_learning_test.py
imports: from absl.testing import absltest
  import jax.numpy
  import numpy
  from rlax._src import model_learning
label_desc: Use image node_id 2 for calling the ModelLearningTest obj's underlying member method code with example usage: obj.test_extract_subsequences() without return types