Merge pull request #628 from jeverink/geometry-printing

Update printing for geometry

Author: chaozg

cuqi/geometry/_geometry.py

@@ -225,7 +225,12 @@ def __eq__(self, obj):
         return self._all_values_equal(obj)
 
     def __repr__(self) -> str:
-        return "{}{}".format(self.__class__.__name__,self.par_shape)
+        if self.par_shape == self.fun_shape:
+            return "{}[{}]".format(self.__class__.__name__,
+                                   self.par_shape if len(self.par_shape) != 1 else self.par_shape[0])
+        return "{}[{}: {}]".format(self.__class__.__name__,
+                                   self.par_shape if len(self.par_shape) != 1 else self.par_shape[0],
+                                   self.fun_shape if len(self.fun_shape) != 1 else self.fun_shape[0])
 
     def _all_values_equal(self, obj):
         """Returns true of all values of the object and self are equal"""