def test_createArray_one():
diffrData = read(h5_file, index=2, poissonize=False)
diffrData = read(h5_file, index="2", poissonize=False)
diffr_pattern = diffrData.array
diffrData = read(h5_file, poissonize=False)
diffr_patterns = diffrData.array
assert len(diffr_pattern) == 1
assert np.array_equal(diffr_patterns[2], diffr_pattern[0]) is True
if __name__ == "__main__":
plt.figure()
plt.imshow(diffr_pattern[0], norm=mpl.colors.LogNorm())
plt.colorbar()
plt.show()
def test_createArray_partial():
diffrData = read(h5_file, index="2:5", poissonize=False)
diffr_patterns = diffrData.array
diffrData = read(h5_file, index=4, poissonize=False)
diffr_pattern = diffrData.array
assert len(diffr_patterns) == 3
assert np.array_equal(diffr_patterns[2], diffr_pattern[0]) is True
if __name__ == "__main__":
fig, ax = plt.subplots(1, 2)
im = ax[0].imshow(diffr_patterns[0],
norm=mpl.colors.LogNorm(vmin=1e-8, vmax=1e-3))
divider = make_axes_locatable(ax[0])
cax = divider.append_axes("right", size="5%", pad=0.1)
fig.colorbar(im, cax=cax)
im = ax[1].imshow(diffr_patterns[1],
norm=mpl.colors.LogNorm(vmin=1e-8, vmax=1e-3))
divider = make_axes_locatable(ax[1])
cax = divider.append_axes("right", size="5%", pad=0.1)
fig.colorbar(im, cax=cax)
fig.tight_layout()
plt.show()
def test_read():
diffrData = read(h5_file, poissonize=False)
diffr_patterns = diffrData.array
assert len(diffr_patterns) == 13
assert np.sum(diffr_patterns) != 0
if __name__ == "__main__":
plt.figure()
plt.imshow(diffr_patterns[0], norm=mpl.colors.LogNorm())
plt.figure()
plt.imshow(diffr_patterns[2], norm=mpl.colors.LogNorm())
plt.colorbar()
plt.show()
def test_Poission():
diffrData = read(h5_file, poissonize=True)
assert np.sum(diffrData.array) == 0
def test_write(tmp_path):
tmp_h5 = str(tmp_path / 'tmp.h5')
write(tmp_h5, read(h5_file), method_desciption='', pmi_file_list=None)