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 )