[test] added test units

bob/ip/pytorch_extractor/test.py

+import pkg_resources
+import numpy
+numpy.random.seed(10)
+import os
+
+
+def test_cnn8():
+
+    from . import CNN8Extractor
+    extractor = CNN8Extractor()
+    
+    # this architecture expects 3x128x128 images
+    data = numpy.random.rand(3, 128, 128).astype("float32")
+    output = extractor(data)
+    assert output.shape[0] == 512
+
+def test_casianet():
+
+    from . import CasiaNetExtractor
+    extractor = CasiaNetExtractor()
+    # this architecture expects 3x128x128 images
+    data = numpy.random.rand(3, 128, 128).astype("float32")
+    output = extractor(data)
+    assert output.shape[0] == 320