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')
