mirror of https://github.com/mahmoud/boltons.git
105 lines
3.7 KiB
Python
105 lines
3.7 KiB
Python
|
|
from boltons.statsutils import Stats
|
|
|
|
|
|
# [round(random.normalvariate(10, 3), 3) for i in range(100)]
|
|
NORM_DATA = [12.975, 8.341, 12.27, 12.799, 15.443, 6.739, 10.572,
|
|
14.863, 3.723, 9.825, 7.716, 12.218, 11.641, 9.02,
|
|
13.037, 11.175, 13.156, 11.706, 8.184, 13.306, 9.845,
|
|
11.665, 14.298, 12.021, 8.419, 10.209, 10.698, 6.559,
|
|
10.346, 9.895, 11.742, 13.391, 10.587, 6.639, 10.23,
|
|
8.841, 10.511, 6.033, 5.767, 8.482, 9.517, 9.039, 11.111,
|
|
13.845, 4.331, 5.323, 14.486, 14.875, 10.005, 6.367,
|
|
12.18, 11.69, 13.97, 4.14, 7.979, 11.114, 4.126, 10.028,
|
|
9.295, 10.078, 14.615, 7.055, 7.641, 9.037, 9.933,
|
|
10.077, 14.174, 14.645, 10.398, 10.238, 9.067, 4.841,
|
|
13.159, 15.829, 8.464, 7.47, 11.858, 9.885, 11.978,
|
|
5.418, 12.19, 8.206, 10.755, 6.455, 10.019, 11.594,
|
|
9.082, 10.245, 12.321, 8.508, 9.711, 5.5, 15.001, 9.922,
|
|
7.864, 7.794, 10.546, 9.203, 8.798, 9.853]
|
|
|
|
|
|
SIMPLE_RANGE_DATA = range(110)
|
|
|
|
LAYER_RANGE_DATA = (list(range(100))
|
|
+ list(range(20, 80))
|
|
+ list(range(40, 60)))
|
|
|
|
EMPTY_DATA = []
|
|
|
|
ALL_DATASETS = [EMPTY_DATA, LAYER_RANGE_DATA, SIMPLE_RANGE_DATA, NORM_DATA]
|
|
|
|
|
|
def test_check_sum():
|
|
for data in ALL_DATASETS:
|
|
for bin_size in [0, 1, 10, 99]:
|
|
# bin_size=0 tests freedman
|
|
stats = Stats(data)
|
|
hist_counts = stats.get_histogram_counts()
|
|
hist_counts_sum = sum([c for _, c in hist_counts])
|
|
assert len(data) == hist_counts_sum
|
|
if not data:
|
|
continue
|
|
|
|
assert min(data) >= hist_counts[0][0]
|
|
assert max(data) >= hist_counts[-1][0]
|
|
return
|
|
|
|
|
|
def test_norm_regression():
|
|
stats = Stats(NORM_DATA)
|
|
|
|
assert stats.format_histogram(width=80) == NORM_DATA_FREEDMAN_OUTPUT
|
|
assert stats.format_histogram(10, width=80) == NORM_DATA_TEN_BIN_OUTPUT
|
|
|
|
subpar_bin_out = stats.format_histogram([12.0], width=80)
|
|
assert subpar_bin_out == NORM_DATA_SINGLE_SUBPAR_BIN_OUTPUT
|
|
|
|
format_bin_out = stats.format_histogram(5,
|
|
width=80,
|
|
format_bin=lambda b: '%sms' % b)
|
|
assert format_bin_out == NORM_DATA_FORMAT_BIN_OUTPUT
|
|
|
|
|
|
NORM_DATA_FREEDMAN_OUTPUT = """\
|
|
3.7: 5 ################
|
|
5.2: 10 ################################
|
|
6.8: 9 #############################
|
|
8.3: 21 ###################################################################
|
|
9.9: 22 ######################################################################
|
|
11.4: 16 ###################################################
|
|
13.0: 10 ################################
|
|
14.5: 7 ######################"""
|
|
|
|
NORM_DATA_TEN_BIN_OUTPUT = """\
|
|
3.7: 5 ##############
|
|
4.9: 5 ##############
|
|
6.1: 6 #################
|
|
7.3: 12 ##################################
|
|
8.5: 11 ###############################
|
|
9.7: 25 ######################################################################
|
|
10.9: 12 ##################################
|
|
12.1: 12 ##################################
|
|
13.4: 5 ##############
|
|
14.6: 7 ####################"""
|
|
|
|
# make sure the minimum gets added
|
|
NORM_DATA_SINGLE_SUBPAR_BIN_OUTPUT = """\
|
|
3.7: 75 ######################################################################
|
|
12.0: 25 #######################"""
|
|
|
|
NORM_DATA_FORMAT_BIN_OUTPUT = """\
|
|
3.7ms: 10 ###################
|
|
6.1ms: 18 ##################################
|
|
8.5ms: 36 ####################################################################
|
|
10.9ms: 24 #############################################
|
|
13.4ms: 12 #######################"""
|
|
|
|
|
|
def main():
|
|
print(Stats(NORM_DATA).format_histogram(10))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|