def test_add_sub_parser(self): # setting up required parsers parser = argparse.ArgumentParser() parent = argparse.ArgumentParser(add_help=False) subparsers = parser.add_subparsers() # adding percentiles subparser HistogramLogscale._add_subparser(subparsers, parent) # testing parser cargs = 'HistogramLogscale'.split() args = parser.parse_args(cargs) assert args.scale_fact == 10.0 # cargs = 'histlog 5'.split() args = parser.parse_args(cargs) assert args.scale_fact == 5.0
def test_define_bins(self, data_field_class): r""" Testing logscale bin generation """ hist = HistogramLogscale(data_field_class()) hist.args = {'scale_fact': 10} hist.data_map = sp.ones((hist.nz, hist.nx), dtype=int) for i in range(hist.nz): hist.data_map[i, :] = 10 * (i+2) hist.data_vector = sp.ravel(hist.data_map) hist.data_vector[0] = -1 # hist.define_bins() # assert len(hist.bins) == 5 assert hist.bins[0][0] == hist.data_vector[0] assert hist.bins[-1][1] > hist.data_vector[-1]