56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
from saucebrush.stats import Sum, Average, Median, MinMax, StandardDeviation, Histogram
|
|
|
|
|
|
def _simple_data():
|
|
return [
|
|
{"a": 1, "b": 2, "c": 3},
|
|
{"a": 5, "b": 5, "c": 5},
|
|
{"a": 1, "b": 10, "c": 100},
|
|
]
|
|
|
|
|
|
def test_sum():
|
|
fltr = Sum("b")
|
|
list(fltr.attach(_simple_data()))
|
|
assert fltr.value() == 17
|
|
|
|
|
|
def test_average():
|
|
fltr = Average("c")
|
|
list(fltr.attach(_simple_data()))
|
|
assert fltr.value() == 36.0
|
|
|
|
|
|
def test_median():
|
|
# odd number of values
|
|
fltr = Median("a")
|
|
list(fltr.attach(_simple_data()))
|
|
assert fltr.value() == 1
|
|
|
|
# even number of values
|
|
fltr = Median("a")
|
|
list(fltr.attach(_simple_data()[:2]))
|
|
assert fltr.value() == 3
|
|
|
|
|
|
def test_minmax():
|
|
fltr = MinMax("b")
|
|
list(fltr.attach(_simple_data()))
|
|
assert fltr.value() == (2, 10)
|
|
|
|
|
|
def test_standard_deviation():
|
|
fltr = StandardDeviation("c")
|
|
list(fltr.attach(_simple_data()))
|
|
assert fltr.average() == 36.0
|
|
assert fltr.median() == 5
|
|
assert fltr.value() == (55.4346462061408, 3073.0)
|
|
assert fltr.value(True) == (45.2621990922521, 2048.6666666666665)
|
|
|
|
|
|
def test_histogram():
|
|
fltr = Histogram("a")
|
|
fltr.label_length = 1
|
|
list(fltr.attach(_simple_data()))
|
|
assert str(fltr) == "\n1 **\n5 *\n"
|