[test] fixed test for LightCNN9Extractor (input data)

bob/ip/pytorch_extractor/test.py

@@ -41,7 +41,7 @@ def test_lightcnn9():
     """
     from . import LightCNN9Extractor
     extractor = LightCNN9Extractor()
-    # this architecture expects 1x128x128 images
-    data = numpy.random.rand(1, 128, 128).astype("float32")
+    # this architecture expects 128x128 grayscale images
+    data = numpy.random.rand(128, 128).astype("float32")
     output = extractor(data)
     assert output.shape[0] == 256