Add 'clean' function

Author: grantjenks

tests/test_coverage.py

@@ -1,7 +1,7 @@
 import os
 import sys
 from .context import wordsegment
-from wordsegment import load, main, isegment, segment, UNIGRAMS, BIGRAMS
+from wordsegment import clean, load, main, isegment, segment, UNIGRAMS, BIGRAMS
 
 load()
 
@@ -11,6 +11,9 @@ def test_unigrams():
 def test_bigrams():
     assert 'in the' in BIGRAMS
 
+def test_clean():
+    assert clean("Can't buy me love!") == 'cantbuymelove'
+
 def test_segment_0():
     result = ['choose', 'spain']
     assert segment(''.join(result)) == result

wordsegment/__init__.py

@@ -175,6 +175,7 @@ def clean(cls, text):
 
 
 _segmenter = Segmenter()        # pylint: disable=invalid-name
+clean = _segmenter.clean        # pylint: disable=invalid-name
 load = _segmenter.load          # pylint: disable=invalid-name
 isegment = _segmenter.isegment  # pylint: disable=invalid-name
 segment = _segmenter.segment    # pylint: disable=invalid-name