code: def test_make_periodic_input_warping_onedim(self):
      iw = kernels.make_periodic_input_warping(4, 0, True)
      np.testing.assert_allclose(
          jnp.array([0, 1, 1, 2, 3, 4, 5]),
          iw(jnp.array([1, 2, 3, 4, 5])),
          atol=1e-6,
      )
      iw = kernels.make_periodic_input_warping(4, 0, False)
      np.testing.assert_allclose(
          jnp.array([0, 1, 2, 3, 4, 5]),
          iw(jnp.array([1, 2, 3, 4, 5])),
          atol=1e-6,
      )
  
imports: from absl.testing import parameterized
  import jax
  import jax.numpy
  import numpy
  from tensorflow_probability.python.experimental.autobnn import kernels
  from tensorflow_probability.python.experimental.autobnn import util
  from tensorflow_probability.substrates.jax.distributions import lognormal
  from absl.testing import absltest
label_desc: Use image node_id 6 for calling the KernelsTest obj's underlying member method code with example usage: obj.test_make_periodic_input_warping_onedim() without return types