docstr: Batch dims have to be the same or 1.
docstr_tok: ["Batch","dims","have","to","be","the","same","or","1","."]
code: def test_initialize_lights_broadcast_fail(self):
      """
      Batch dims have to be the same or 1.
      """
      with self.assertRaises(ValueError):
          DirectionalLights(
              ambient_color=torch.randn(10, 3),
              diffuse_color=torch.randn(15, 3),
          )
  
      with self.assertRaises(ValueError):
          PointLights(
              ambient_color=torch.randn(10, 3),
              diffuse_color=torch.randn(15, 3),
          )
  
imports: import unittest
  import numpy
  import torch
  from pytorch3d.renderer.lighting import AmbientLights, DirectionalLights, PointLights
  from pytorch3d.transforms import RotateAxisAngle
  from .common_testing import TestCaseMixin
label_desc: Use image node_id 5 for calling the TestLights obj's underlying member method code with example usage: obj.test_initialize_lights_broadcast_fail() without return types