tvii: tests/test_centroid.py comparison

add test for centroid calculations

comparison

equal deleted inserted replaced

-:02f586a9defe
+:20a2970d4510
+#!/usr/bin/env python
+"""
+test centroid calculations
+"""
+import common
+import os
+import unittest
+from tvii import centroid
+class CentroidTest(unittest.TestCase):
+def test_square(self):
+"""find the centroids of these boring squares"""
+# a boring square centered about the origin
+points = [(5., 0.),
+(0., 5.),
+(-5., 0.),
+(0., -5)]
+center = centroid.centroid(*points)
+tolerance = 1e-6
+assert len(center) == 2  # dimensions
+assert all([abs(dim) < tolerance
+for dim in center])
+def test_main(self):
+"""smoketest for console-script entry point"""
+datafile = common.datafile('circles.csv')
+assert os.path.exists(datafile)
+centroid.main([datafile])
+if __name__ == '__main__':
+unittest.main()

Mercurial > hg > tvii