Merge pull request #2786 from andrehora/master

improving tests in test_cells and test_pick

Author: willmcgugan

CONTRIBUTORS.md

@@ -18,6 +18,7 @@ The following people have contributed to the development of Rich:
 - [Michał Górny](https://github.com/mgorny)
 - [Nok Lam Chan](https://github.com/noklam)
 - [Leron Gray](https://github.com/daddycocoaman)
+- [Andre Hora](https://github.com/andrehora)
 - [Kenneth Hoste](https://github.com/boegel)
 - [Lanqing Huang](https://github.com/lqhuang)
 - [Finn Hughes](https://github.com/finnhughes)

tests/test_cells.py

@@ -4,6 +4,15 @@
 def test_cell_len_long_string():
     # Long strings don't use cached cell length implementation
     assert cells.cell_len("abc" * 200) == 3 * 200
+    # Boundary case
+    assert cells.cell_len("a" * 512) == 512
+
+
+def test_cell_len_short_string():
+    # Short strings use cached cell length implementation
+    assert cells.cell_len("abc" * 100) == 3 * 100
+    # Boundary case
+    assert cells.cell_len("a" * 511) == 511
 
 
 def test_set_cell_size():

tests/test_pick.py

@@ -2,6 +2,9 @@
 
 
 def test_pick_bool():
+    assert pick_bool(False) == False
+    assert pick_bool(True) == True
+    assert pick_bool(None) == False
     assert pick_bool(False, True) == False
     assert pick_bool(None, True) == True
     assert pick_bool(True, None) == True