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
|
#!/usr/bin/env python
from cogent.util.unit_test import TestCase, main
from cogent.maths.stats.information_criteria import aic, bic
__author__ = "Gavin Huttley"
__copyright__ = "Copyright 2007-2012, The Cogent Project"
__credits__ = ["Gavin Huttley"]
__license__ = "GPL"
__version__ = "1.5.3"
__maintainer__ = "Gavin Huttley"
__email__ = "gavin.huttley@anu.edu.au"
__status__ = "Production"
class InformationCriteria(TestCase):
"""Tests calculation of AIC and BIC measures."""
def test_aic(self):
"""correctly compute AIC from Burnham & Anderson 2002, p102"""
self.assertFloatEqual(aic(-9.7039, 4), 27.4078)
def test_aic_corrected(self):
"""correctly compute AIC corrected for small sample size"""
# from Burnham & Anderson 2002, p102
self.assertFloatEqual(aic(-9.7039, 4, sample_size=13), 32.4078)
def test_bic(self):
"""correctly compute BIC"""
# against hand calculated
self.assertFloatEqual(bic(-9.7039, 4, 13), 29.6675974298)
if __name__ == "__main__":
main()
|
