def METHOD_NAME(self):
    """
    Batch dims have to be the same or 1.
    """
    with self.assertRaises(ValueError):
        Materials(
            ambient_color=torch.randn(10, 3), diffuse_color=torch.randn(15, 3)
        )