code: def test_reflection_conv():
      array = jnp.zeros((5, 6))
      array = array.at[2, 3].set(1.0)
      kernel = jnp.ones((2, 3))
  
      output = conv.reflection_conv(kernel, array)
      true_output = jnp.array(
          [
              [
                  0.0,
                  0.0,
                  0.0,
                  0.0,
                  0.0,
                  0.0,
              ],
              [
                  0.0,
                  0.0,
                  1.0,
                  1.0,
                  1.0,
                  0.0,
              ],
              [
                  0.0,
                  0.0,
                  1.0,
                  1.0,
                  1.0,
                  0.0,
              ],
              [
                  0.0,
                  0.0,
                  0.0,
                  0.0,
                  0.0,
                  0.0,
              ],
          ]
      )
      assert jnp.allclose(output, true_output)
  
imports: import numpy
  from jax import numpy
  from pytest import mark
  from jaxdf import conv
label_desc: Use image node_id 1 for calling a global function with example usage: test_reflection_conv() without return types