docstr: Color should have shape (N, 3) or (1, 3)
docstr_tok: ["Color","should","have","shape","(","N",",","3",")","or","(","1",",","3",")"]
code: def test_initialize_lights_dimensions_fail(self):
      """
      Color should have shape (N, 3) or (1, 3)
      """
      with self.assertRaises(ValueError):
          DirectionalLights(ambient_color=torch.randn(10, 4))
  
      with self.assertRaises(ValueError):
          DirectionalLights(direction=torch.randn(10, 4))
  
      with self.assertRaises(ValueError):
          PointLights(ambient_color=torch.randn(10, 4))
  
      with self.assertRaises(ValueError):
          PointLights(location=torch.randn(10, 4))
  
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 6 for calling the TestLights obj's underlying member method code with example usage: obj.test_initialize_lights_dimensions_fail() without return types