Correction of infer_shape tests

theano/tensor/tests/test_extra_ops.py

@@ -736,20 +736,35 @@ class test_Unique(utt.InferShapeTester):
         Testing the infer_shape with a vector.
         """
         x = theano.tensor.vector()
-        for op in self.ops :
+
+        for op in self.ops:
+            if not op.return_inverse:
+                continue
+            if op.return_index :
+                f = op(x)[2]
+            else:
+                f = op(x)[1]
             self._compile_and_check([x],  
-                                    [op(x)], 
+                                    [f], 
                                     [np.asarray(np.array([2,1,3,2]),
                                                 dtype=config.floatX)],
                                     self.op_class)
         
     def test_infer_shape_matrix(self):                  
         """
-        Testing the infer_shape with a vector.
+        Testing the infer_shape with a matrix.
         """
         x = theano.tensor.matrix()
+        
+        for op in self.ops:
+            if not op.return_inverse:
+                continue
+            if op.return_index :
+                f = op(x)[2]
+            else:
+                f = op(x)[1]
             self._compile_and_check([x],  
-                                [self.op(x)], 
+                                [f], 
                                 [np.asarray(np.array([[2, 1], [3, 2],[2, 3]]),
                                             dtype=config.floatX)],
                                 self.op_class)