def test_distHistoReader_first_last():
reader = postproc_io.distHistoReader(directory=out_directory,
first=2,
last=8,
verbose=False)
writer = postproc_io.DummyWriter(source=reader)
writer.dump()
def test_average_filter_all():
n_tot = 100
n_bins = 1024
reader = postproc_io.DummyReader(n_histogram_sets=n_tot,
n_bins=n_bins,
random=False)
average = postproc_filter.Average(source=reader, n_avg='all')
counter = FrameCounter(source=average)
keeper = KeepLastFrame(source=counter)
writer = postproc_io.DummyWriter(source=keeper)
writer.dump()
# consistency checks
assert (counter.count == 1)
# check the last processed frame
frm = keeper.last_frame
keys = frm.get_keys(base.loc_histograms, skip_keys='radii')
for key in keys:
histo = frm.get_data(base.loc_histograms + '/' + key)
assert (np.sum(histo, dtype=np.int) == n_bins)
def test_average_filter():
n_tot = 100
n_avg = 10
n_bins = 1024
# create histograms filled with ones
reader = postproc_io.DummyReader(n_histogram_sets=n_tot,
n_bins=n_bins,
random=False)
average = postproc_filter.Average(source=reader, n_avg=n_avg)
counter = FrameCounter(source=average)
keeper = KeepLastFrame(source=counter)
writer = postproc_io.DummyWriter(source=keeper)
writer.dump()
# consistency checks
frame_indices_expected = np.arange(n_avg, n_tot + 1, n_avg)
frame_indices_measured = np.array(counter.frames)
assert (np.isclose(counter.count, n_tot / n_avg))
assert np.allclose(frame_indices_measured, frame_indices_expected)
# check the last processed frame
frm = keeper.last_frame
keys = frm.get_keys(base.loc_histograms, skip_keys='radii')
for key in keys:
histo = frm.get_data(base.loc_histograms + '/' + key)
assert (np.sum(histo, dtype=np.int) == n_bins)
def test_H5Reader():
reader = postproc_io.H5Reader(file=h5name, verbose=False)
writer = postproc_io.DummyWriter(source=reader)
writer.dump()
def test_distHistoReader_step():
reader = postproc_io.distHistoReader(directory=out_directory,
step=2,
verbose=False)
writer = postproc_io.DummyWriter(source=reader)
writer.dump()
def test_DummyReader_DummyWriter():
reader = postproc_io.DummyReader()
writer = postproc_io.DummyWriter(source=reader)
writer.dump()
def test_dummy_filter():
reader = postproc_io.DummyReader()
filtre = postproc_filter.Dummy(source=reader)
writer = postproc_io.DummyWriter(source=filtre)
writer.dump()