Remove plt.show calls in tests