Back to 100% tests coverage.

wireservice · Oct 30, 2015 · d8d61cb · d8d61cb
1 parent a842a02
commit d8d61cb
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -410,6 +410,25 @@ def test_order_by(self):
         self.assertSequenceEqual(table.rows[1], (2, 3, 'b'))
         self.assertSequenceEqual(table.rows[2], (None, 2, u'👍'))
 
+    def test_order_by_nulls(self):
+        self.rows = (
+            (1, 4, 'a'),
+            (None, 3, 'c'),
+            (2, 3, 'b'),
+            (None, 2, u'👍')
+        )
+
+        table = Table(self.rows, self.columns)
+
+        new_table = table.order_by('one')
+
+        self.assertIsNot(new_table, table)
+        self.assertEqual(len(new_table.rows), 3)
+        self.assertSequenceEqual(new_table.rows[0], (1, 4, 'a'))
+        self.assertSequenceEqual(new_table.rows[1], (2, 3, 'b'))
+        self.assertSequenceEqual(new_table.rows[2], (None, 3, 'c'))
+        self.assertSequenceEqual(new_table.rows[3], (None, 2, u'👍'))
+
     def test_order_by_func(self):
         rows = (
             (1, 2, 'a'),
@@ -639,6 +658,19 @@ def test_counts_text(self):
 
         self.assertSequenceEqual(new_table.row_names, ['Y', 'N', None])
 
+    def test_counts_key_func(self):
+        table = Table(self.rows, self.columns)
+        new_table = table.counts(lambda r: r['two'])
+
+        self.assertEqual(len(new_table.rows), 3)
+        self.assertEqual(len(new_table.columns), 2)
+
+        self.assertSequenceEqual(new_table.rows[0], ['Y', 1])
+        self.assertSequenceEqual(new_table.rows[1], ['N', 4])
+        self.assertSequenceEqual(new_table.rows[2], [None, 1])
+
+        self.assertSequenceEqual(new_table.row_names, ['Y', 'N', None])
+
 class TestBins(unittest.TestCase):
     def setUp(self):
         self.number_type = Number()