diff --git a/tests/test_non_sparse.py b/tests/test_non_sparse.py new file mode 100644 index 000000000..a7a05adf1 --- /dev/null +++ b/tests/test_non_sparse.py @@ -0,0 +1,25 @@ +import py.test + +from spacy.orth import non_sparse +import math + + +def test_common_case_upper(): + cases = {'upper': 0.7, 'lower': 0.2, 'title': 0.1} + prob = math.log(0.1) + assert non_sparse('usa', prob, 0, cases, {}) == 'USA' + +def test_same(): + cases = {'upper': 0.01, 'title': 0.09, 'lower': 0.9} + prob = math.log(0.5) + assert non_sparse('the', prob, 0, cases, {}) == 'the' + +def test_common_case_lower(): + prob = math.log(0.5) + cases = {'upper': 0.01, 'title': 0.09, 'lower': 0.9} + assert non_sparse('The', prob, 0, cases, {}) == 'the' + +def test_shape(): + prob = math.log(0.00001) + cases = {'upper': 0.0, 'title': 0.0, 'lower': 0.0} + assert non_sparse('1999', prob, 0, cases, {}) == 'dddd'