Plotting notebook: added plt.figure() calls

in unit testing without display plots, the same figure would be used
over and over again. This is problematic when mixing 3D and 2D axes
7 jobs for master in 5 minutes and 24 seconds (queued for 7 minutes and 28 seconds)