1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
#!/usr/bin/env python
"""Provides tests for Histogram.
"""
from cogent.util.unit_test import TestCase, main
from cogent.maths.stats.histogram import Histogram
from cogent.core.location import Span
__author__ = "Sandra Smit"
__copyright__ = "Copyright 2007-2012, The Cogent Project"
__credits__ = ["Rob Knight", "Sandra Smit"]
__license__ = "GPL"
__version__ = "1.5.3"
__maintainer__ = "Sandra Smit"
__email__ = "sandra.smit@colorado.edu"
__status__ = "Production"
class HistogramTests(TestCase):
"""Tests for Histogram class"""
def test_init_no_bins(self):
"""Histogram should raise error if initialized without bins"""
# you deserve an Error if you initialize your histogram
# without providing Bins
self.assertRaises(AttributeError, Histogram)
def test_init_bins(self):
"""Histogram should set _bins property correctly"""
bins = [Span(0,2),Span(2,4),Span(4,6)]
bins_only = Histogram(bins=bins)
self.assertEqual(bins_only._bins, bins)
def test_init_bins_data(self):
"""Histogram should fill bins with data if supplied"""
# most basic histogram, bins and data
data = [1,3,5,'A']
bins = [Span(0,2),Span(2,4),Span(4,6)]
data_and_bins = Histogram(data=data,bins=bins)
self.assertEqual(data_and_bins._bins,bins)
self.assertEqual(data_and_bins._values,[[1],[3],[5]])
self.assertEqual(data_and_bins.Other,['A'])
def test_call(self):
"""Histogram __call__ should update with new data"""
data = [1,3,5,'A']
bins = [Span(0,2),Span(2,4),Span(4,6)]
data_and_bins = Histogram(data=data,bins=bins)
#update the histogram
data_and_bins([4,5,6,7])
self.assertEqual(data_and_bins._values,[[1],[3],[5,4,5]])
self.assertEqual(data_and_bins.Other,['A',6,7])
def test_mapping(self):
"""Histogram Mapping should apply correct function to values"""
# bins, data, mapping
data = ['A','AAA','CCCCC','GGGGGGGGGGGGGG']
bins = [Span(0,2),Span(2,4),Span(4,6)]
mapping = Histogram(data=data,bins=bins,Mapping=len)
self.assertEqual(mapping._values, [['A'],['AAA'],['CCCCC']])
self.assertEqual(mapping.Other,['GGGGGGGGGGGGGG'])
def test_multi(self):
"""Histogram Multi should allow values to match multiple bins"""
#bins, data, multi=True
bins2 = [Span(0,5),Span(3,8),Span(6,10)]
data2 = [0,1,2,3,4,5,6,7,8,9,10]
not_multi = Histogram(data2,bins2)
self.assertEqual(not_multi._values,[[0,1,2,3,4],[5,6,7],[8,9]])
self.assertEqual(not_multi.Other,[10])
multi = Histogram(data2,bins2,Multi=True)
self.assertEqual(multi._values,[[0,1,2,3,4],[3,4,5,6,7],[6,7,8,9]])
self.assertEqual(multi.Other,[10])
def test_toFreqs(self):
"""Histogram toFreqs() should return a Freqs object"""
h = Histogram(range(0,20),bins=[Span(0,3),Span(3,10),
Span(10,18),Span(18,20)])
constructor=str
f = h.toFreqs()
self.assertEqual(f[constructor(Span(0,3))],3)
self.assertEqual(f[constructor(Span(3,10))],7)
self.assertEqual(f[constructor(Span(10,18))],8)
self.assertEqual(f[constructor(Span(18,20))],2)
def test_clear(self):
"""Histogram clear should reset all data"""
data = [1,3,5,'A']
bins = [Span(0,2),Span(2,4),Span(4,6)]
data_and_bins = Histogram(data=data,bins=bins)
self.assertEqual(data_and_bins._bins,bins)
self.assertEqual(data_and_bins._values,[[1],[3],[5]])
self.assertEqual(data_and_bins.Other,['A'])
data_and_bins.clear()
self.assertEqual(data_and_bins._values,[[],[],[]])
self.assertEqual(data_and_bins.Other,[])
if __name__ == '__main__':
main()
|
