code: def setUp(self):
      super().setUp()
      self.data = np.array([[0.0, 0.0], [7.5, 1.0], [40.0, 40.0]])
      self.query_points = np.array([[2.0, 1.3], [7.5, 0.0]])
  
file_path: rlax/rlax/_src/episodic_memory_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import chex
  import jax
  import numpy
  from rlax._src import episodic_memory
label_desc: Use image node_id 1 for calling the KNNQueryTest obj's underlying member method code with example usage: obj.setUp() without return types