class TestQuantitativeBinning(TestCase):
"""Test the quantitative binning functions"""
MIN_BIN_SIZE = 1
DEFAULT_BINS = 50
def setUp(self):
self.dimension = QuantitativeDimension(
key='fake',
name='fake',
description='fake',
field_name='fake',
default_bins=self.DEFAULT_BINS,
min_bin_size=self.MIN_BIN_SIZE
)
def test_narrow_distribution(self):
min_val = 0
max_val = 11
bin_size = self.dimension._get_bin_size(min_val, max_val, self.DEFAULT_BINS)
self.assertEquals(bin_size, 1)
self.assertEquals(self.dimension._bin_value(min_val, bin_size), 0)
self.assertEquals(self.dimension._bin_value(max_val, bin_size), 11)
def test_wide_distribution(self):
min_val = 1
max_val = 101
bin_size = self.dimension._get_bin_size(min_val, max_val, 5)
self.assertEquals(bin_size, 20)
self.assertEquals(self.dimension._bin_value(min_val, bin_size), 0)
self.assertEquals(self.dimension._bin_value(max_val, bin_size), 100)
def setUp(self):
self.dimension = QuantitativeDimension(
key='fake',
name='fake',
description='fake',
field_name='fake',
default_bins=self.DEFAULT_BINS,
min_bin_size=self.MIN_BIN_SIZE
)
