saucebrush/saucebrush/tests/stats.py

import unittest
from saucebrush.stats import Sum, Average, Median, MinMax, StandardDeviation, Histogram

class StatsTestCase(unittest.TestCase):

    def _simple_data(self):
        return  [{'a':1, 'b':2, 'c':3},
                 {'a':5, 'b':5, 'c':5},
                 {'a':1, 'b':10, 'c':100}]

    def test_sum(self):
        fltr = Sum('b')
        list(fltr.attach(self._simple_data()))
        self.assertEqual(fltr.value(), 17)
    
    def test_average(self):
        fltr = Average('c')
        list(fltr.attach(self._simple_data()))
        self.assertEqual(fltr.value(), 36.0)
    
    def test_median(self):
        # odd number of values
        fltr = Median('a')
        list(fltr.attach(self._simple_data()))
        self.assertEqual(fltr.value(), 1)
        
        # even number of values
        fltr = Median('a')
        list(fltr.attach(self._simple_data()[:2]))
        self.assertEqual(fltr.value(), 3)
        
    def test_minmax(self):
        fltr = MinMax('b')
        list(fltr.attach(self._simple_data()))
        self.assertEqual(fltr.value(), (2, 10))
    
    def test_standard_deviation(self):
        fltr = StandardDeviation('c')
        list(fltr.attach(self._simple_data()))
        self.assertEqual(fltr.average(), 36.0)
        self.assertEqual(fltr.median(), 5)
        self.assertEqual(fltr.value(), (55.4346462061408, 3073.0))
        self.assertEqual(fltr.value(True), (45.2621990922521, 2048.6666666666665))
    
    def test_histogram(self):
        fltr = Histogram('a')
        fltr.label_length = 1
        list(fltr.attach(self._simple_data()))
        self.assertEqual(str(fltr), "1 **\n5 *\n")

if __name__ == '__main__':
    unittest.main()