def test_movie_writer_dpi_default(): # Test setting up movie writer with figure.dpi default. fig = plt.figure() filename = "unused.null" fps = 5 codec = "unused" bitrate = 1 extra_args = ["unused"] def run(): pass writer = animation.MovieWriter(fps, codec, bitrate, extra_args) writer._run = run writer.setup(fig, filename) assert writer.dpi == fig.dpi
ax3.set_ylabel("Nion at x=0") ax4.set_ylabel("Potential at x=0") Efield = -potential.grad()[0] l1 = ax1.plot(x, Efield, c='grey') l2 = ax1_2.plot(x, potential.value, c='tab:green') l3 = ax2.plot(x, Pion.value, label='Positive', c='tab:red') l4 = ax2.plot(x, Nion.value, label='Negative', c='tab:blue') ax3.scatter(step, Nion.value[0], c='r', s=1) ax4.scatter(step, potential.value[0], c='b', s=1) Efield_save[:, step] = Efield potential_save[:, step] = potential.value Pion_save[:, step] = Pion.value Nion_save[:, step] = Nion.value ax2.legend(loc="upper right") fig.suptitle('Step: ' + str(step)) fig.show() plt.pause(0.05) # plt.savefig('Test Image Saver/TEST_'+str(step)+'.png', dpi=1200) ani = animation.MovieWriter(fig, "TestingMovie") # print(np.sum(Nion.value)) for l in [l1, l2, l3, l4]: l[0].remove() # np.savetxt('Pion_1.txt', Pion_save, delimiter='\t') # np.savetxt('Nion_1.txt', Nion_save, delimiter='\t') # np.savetxt('Electrical field intensity_1.txt',Efield_save, delimiter='\t') # np.savetxt('potential.txt_1', potential_save, delimiter='\t')