code: def test_lhs_broadcasting(
      self, source_shape, target_shape, expected_result_shape
  ):
      source = jnp.ones(shape=source_shape, dtype=jnp.float32)
      target = jnp.ones(shape=target_shape, dtype=jnp.float32)
      expected_result = jnp.ones(
          shape=expected_result_shape, dtype=jnp.float32
      )
      result = base.lhs_broadcast(source, target)
      np.testing.assert_array_almost_equal(result, expected_result)
  
file_path: rlax/rlax/_src/base_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  import jax.numpy
  import numpy
  from rlax._src import base
label_desc: Use image node_id 1 for calling the BroadcastTest obj's underlying member method code with example usage: obj.test_lhs_broadcasting(source_shape, target_shape, expected_result_shape) without return types