create dummy versions of trplot functions if matplotlib not installed

petercorke · petercorke · commit 149f7345016f · 2020-05-11T07:37:24.000+10:00
diff --git a/spatialmath/base/transforms2d.py b/spatialmath/base/transforms2d.py
@@ -429,7 +429,7 @@ def trplot2(T, ax=None, dims=None, color='blue', frame=None, textcolor=None, lab
             ax.quiver(o[0], o[1], x[0]-o[0], x[1]-o[1], angles='xy', scale_units='xy', scale=1, linewidth=width, facecolor=color, edgecolor=color)
             ax.quiver(o[0], o[1], y[0]-o[0], y[1]-o[1], angles='xy', scale_units='xy', scale=1, linewidth=width, facecolor=color, edgecolor=color)
             # plot an invisible point at the end of each arrow to allow auto-scaling to work
-            ax.scatter( x=[o[0], x[0], y[0]], y=[o[1], x[1], y[1]], s=[20,0,0,0])
+            ax.scatter( x=[o[0], x[0], y[0]], y=[o[1], x[1], y[1]], s=[20,0,0])
         else:
             ax.plot([o[0], x[0]], [o[1], x[1]], color=color, linewidth=width)
             ax.plot([o[0], y[0]], [o[1], y[1]], color=color, linewidth=width)
@@ -450,8 +450,10 @@ def trplot2(T, ax=None, dims=None, color='blue', frame=None, textcolor=None, lab
     
             ax.text(x[0], x[1], "$%c_{%s}$" % (labels[0],frame), color=color, horizontalalignment='center', verticalalignment='center')
             ax.text(y[0], y[1], "$%c_{%s}$" % (labels[1], frame),  color=color, horizontalalignment='center', verticalalignment='center')
-except:
-    pass
+            
+except:  # pragma: no cover
+    def trplot2(*args, **kwargs):
+        print('** trplot2: no plot produced -- matplotlib not installed')
 
 if __name__ == '__main__':  # pragma: no cover
     import pathlib
diff --git a/spatialmath/base/transforms3d.py b/spatialmath/base/transforms3d.py
@@ -1249,8 +1249,10 @@ def trplot(T, axes=None, dims=None, color='blue', frame=None, textcolor=None, la
             ax.text(x[0], x[1], x[2], "$%c_{%s}$" % (labels[0],frame), color=color, horizontalalignment='center', verticalalignment='center')
             ax.text(y[0], y[1], y[2], "$%c_{%s}$" % (labels[1], frame),  color=color, horizontalalignment='center', verticalalignment='center')
             ax.text(z[0], z[1], z[2], "$%c_{%s}$" % (labels[2], frame),  color=color, horizontalalignment='center', verticalalignment='center')
-except:
-    pass
+
+except:  # pragma: no cover
+    def trplot(*args, **kwargs):
+        print('** trplot: no plot produced -- matplotlib not installed')
     
 if __name__ == '__main__':  # pragma: no cover
     import pathlib
