Test and refactor Numbers.

1 files changed, 35 insertions, 0 deletions

diff --git a/test/test_results.py b/test/test_results.py
new file mode 100644
index 00000000..d70a3b82
--- /dev/null
+++ b/test/test_results.py
@@ -0,0 +1,35 @@
+"""Tests for Coverage.py's results analysis."""
+
+import os, sys
+
+from coverage.results import Numbers
+
+sys.path.insert(0, os.path.split(__file__)[0]) # Force relative import for Py3k
+from coveragetest import CoverageTest
+
+
+class NumbersTest(CoverageTest):
+    """Tests for Coverage.py's numeric measurement summaries."""
+    def test_basic(self):
+        n1 = Numbers(n_files=1, n_statements=200, n_missing=20)
+        self.assertEqual(n1.n_statements, 200)
+        self.assertEqual(n1.n_run, 180)
+        self.assertEqual(n1.percent_covered, 90)
+        
+    def test_addition(self):
+        n1 = Numbers(n_files=1, n_statements=200, n_missing=20)
+        n2 = Numbers(n_files=1, n_statements=10, n_missing=8)
+        n3 = n1 + n2
+        self.assertEqual(n3.n_files, 2)
+        self.assertEqual(n3.n_statements, 210)
+        self.assertEqual(n3.n_run, 182)
+        self.assertAlmostEqual(n3.percent_covered, 86.666666666)
+
+    def test_sum(self):
+        n1 = Numbers(n_files=1, n_statements=200, n_missing=20)
+        n2 = Numbers(n_files=1, n_statements=10, n_missing=8)
+        n3 = sum([n1, n2])
+        self.assertEqual(n3.n_files, 2)
+        self.assertEqual(n3.n_statements, 210)
+        self.assertEqual(n3.n_run, 182)
+        self.assertAlmostEqual(n3.percent_covered, 86.666666666)