code: def test_make_periodic_input_warping_onedim_features(self):
      iw = kernels.make_periodic_input_warping(4, 0, True)
      np.testing.assert_allclose(
          jnp.array(
              [[1, 0, 0], [0, 1, 1], [-1, 0, 2], [0, -1, 3], [1, 0, 4]]
          ),
          iw(jnp.array([[0], [1], [2], [3], [4]])),
          atol=1e-6,
      )
      iw = kernels.make_periodic_input_warping(4, 0, False)
      np.testing.assert_allclose(
          jnp.array([[1, 0], [0, 1], [-1, 0], [0, -1], [1, 0]]),
          iw(jnp.array([[0], [1], [2], [3], [4]])),
          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 7 for calling the KernelsTest obj's underlying member method code with example usage: obj.test_make_periodic_input_warping_onedim_features() without return types