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