From bf474293687231c3a819cd61099e08de686d6af8 Mon Sep 17 00:00:00 2001
From: Matthew Honnibal <matthew@honnibal.com>
Date: Mon, 1 Sep 2014 23:27:31 +0200
Subject: [PATCH] * Add tests for non_sparse string transform

---
 tests/test_non_sparse.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 tests/test_non_sparse.py

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'