accept a mode

examples/simple_plot.py

@@ -3,6 +3,7 @@
 #   a simple matplotlib plot - testing for pipelines
 #
 _version = "3-sep-2023"
+_mode = 0
 
 import os
 import sys
@@ -12,12 +13,21 @@
 
 
 if __name__ == "__main__":
+    if len(sys.argv) > 1:
+        _mode = int(sys.argv[1])
+        plotfile = 'simple_plot.png'
+    else:
+        _mode = 0
+        plotfile = None
+
     x = np.arange(0,2,0.1)
     y = np.sqrt(x)
 
     plt.figure()
     plt.plot(x,y, label="test");
     plt.legend()
-    plt.show()
-    fn = 'simple_plot.png'
-    plt.savefig(fn)
+    if plotfile == None:
+        plt.show()
+    else:
+        plt.savefig(plotfile)
+        print("Wrote ",plotfile)