def test_process_data(self, data_field_class):
r"""
running through the histogram data processing
"""
hist = Histogram(data_field_class())
hist.args = {'num_bins': 10}
hist.data_map = sp.ones((hist.nz, hist.nx), dtype=int)
for i in range(hist.nz):
hist.data_map[i, :] = 1 * (i+1)
hist.data_vector = sp.ravel(hist.data_map)
hist.bins = [(0, 2), (2, 4), (4, 6), (6, 8), (8, 10.01)]
#
hist._process_data()
for bounds, data_bin in zip(hist.bins, hist.processed_data):
assert len(data_bin) == 3
assert data_bin[0] == bounds[0]
assert data_bin[1] == bounds[1]
assert data_bin[2] == 10
#
# testing perserve bins
hist.data_vector = sp.array([-1, 1, 2, 3])
hist.bins = [(0, 5), (5, 10)]
hist._process_data(preserve_bins=True)
assert len(hist.bins) == 2
def test_define_bins(self, data_field_class):
r"""
Testing bin generation
"""
hist = Histogram(data_field_class())
hist.args = {'num_bins': 10}
hist.nz = 100
hist.nx = 100
hist.data_map = sp.ones((hist.nz, hist.nx), dtype=int)
for i in range(hist.nz):
hist.data_map[i, :] = 1 * (i+1)
hist.data_map[0] = -1
hist.data_vector = sp.ravel(hist.data_map)
#
hist.define_bins()
assert len(hist.bins) == hist.args['num_bins'] + 1
