class TestEmptyFrequencyArray(unittest.TestCase):
def setUp(self):
self.empty_fa = FrequencyArray()
self.ranged_fa = FrequencyArray()
for i in range(10, 0, -1):
self.ranged_fa.add(i)
def testNoDataToStart(self):
self.assertEqual(self.empty_fa.data, [])
def testAddData(self):
self.empty_fa.add(1)
self.empty_fa.add(3)
self.assertEqual(self.empty_fa.data, [1, 3])
def testAddDataWithCounts(self):
self.empty_fa.add(1, 2)
self.empty_fa.add(3)
self.assertEqual(self.empty_fa.data, [1, 1, 3])
def testOrderedData(self):
self.assertEqual(self.ranged_fa.orderedData(), range(1, 11))
def testMax(self):
self.assertEqual(self.ranged_fa.max, 10)
def testMin(self):
self.assertEqual(self.ranged_fa.min, 1)
class TestHistogram(unittest.TestCase):
def setUp(self):
self.ranged_fa = FrequencyArray()
self.ranged_ft = FrequencyTable()
self.expected_all = {}
for i in range(10, 0, -1):
self.ranged_fa.add(i)
self.ranged_ft.add(i)
self.expected_all[i] = 1
def testRangeToHistogram(self):
histogram = Histogram(1)
for i in range(10, 0, -1):
histogram.add(i)
self.assertEqual(histogram.data, self.expected_all)
def testArrayToHistogram(self):
histogram = Histogram(1)
for i in self.ranged_fa.explode():
histogram.add(i)
self.assertEqual(histogram.data, self.expected_all)
def testTableToHistogram(self):
histogram = Histogram(1)
for i in self.ranged_ft.explode():
histogram.add(i)
self.assertEqual(histogram.data, self.expected_all)
def testHistogramBucketWidthOfTwo(self):
histogram = Histogram(2)
for i in range(10, 0, -1):
histogram.add(i)
self.assertEqual(histogram.data, {0:1, 1:2, 2:2, 3:2, 4:2, 5:1})
def testHistogramBucketWidthOfTwo(self):
histogram = Histogram(2)
for i in range(10, 0, -1):
histogram.add(i)
self.assertEqual(histogram.orderedData(), [(0,1), (2,2), (4,2), (6,2), (8,2), (10,1)])
def testHistogramBucketWidthOfThree(self):
histogram = Histogram(3)
for i in range(10, 0, -1):
histogram.add(i)
self.assertEqual(histogram.orderedData(), [(0,2), (3,3), (6,3), (9,2)])
def setUp(self):
self.ranged_fa = FrequencyArray()
self.ranged_ft = FrequencyTable()
self.expected_all = {}
for i in range(10, 0, -1):
self.ranged_fa.add(i)
self.ranged_ft.add(i)
self.expected_all[i] = 1
def setUp(self):
self.empty_fa = FrequencyArray()
self.ranged_fa = FrequencyArray()
for i in range(10, 0, -1):
self.ranged_fa.add(i)
