add more tests

4 files changed, 19 insertions, 8 deletions

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 152c5f1..1d2b7c6 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -2,6 +2,7 @@
   - Including certain files (e.g. license.md) in sdists via MANIFEST.in (@proinsias)
   - Relax licensing by moving from BSD to MIT
   - Add Python 3.5 support
+  - Add more tests
 
 ## 1.2.0
 
diff --git a/pep8.sh b/pep8.sh
index f48f800..a8e960a 100755
--- a/pep8.sh
+++ b/pep8.sh
@@ -1,14 +1,11 @@
 #!/bin/bash
 
-echo -e "\nRunning: pep8 ... \n"
-
 # Ignoring autogenerated files
 #  -- Migration directories
 # Ignoring error codes
 #  -- E128 continuation line under-indented for visual indent
+#  -- E261 at least two spaces before inline comment
 #  -- E225 missing whitespace around operator
 #  -- E501 line too long
 
-pep8 --ignore=E128,E225,E501 slugify test.py setup.py
-
-echo -e "Done.\n"
+pep8 --ignore=E128,E261,E225,E501 slugify test.py setup.py
diff --git a/slugify/slugify.py b/slugify/slugify.py
index 113afdd..327f2c1 100644
--- a/slugify/slugify.py
+++ b/slugify/slugify.py
@@ -16,7 +16,7 @@ except ImportError:
 import unidecode
 
 
-__all__ = ['slugify']
+__all__ = ['slugify', 'smart_truncate']
 
 
 CHAR_ENTITY_PATTERN = re.compile('&(%s);' % '|'.join(name2codepoint))
@@ -65,7 +65,7 @@ def smart_truncate(string, max_length=0, word_boundaries=False, separator=' ', s
             else:
                 if save_order:
                     break
-    if not truncated:
+    if not truncated: # pragma: no cover
         truncated = string[:max_length]
     return truncated.strip(separator)
 
@@ -152,7 +152,7 @@ def slugify(text, entities=True, decimal=True, hexadecimal=True, max_length=0, w
     return text
 
 
-def main():
+def main(): # pragma: no cover
     if len(sys.argv) < 2:
         print("Usage %s TEXT TO SLUGIFY" % sys.argv[0])
     else:
diff --git a/test.py b/test.py
index 5bad955..768fcb9 100644
--- a/test.py
+++ b/test.py
@@ -2,6 +2,7 @@
 
 import unittest
 from slugify import slugify
+from slugify import smart_truncate
 
 
 class TestSlugification(unittest.TestCase):
@@ -167,5 +168,17 @@ class TestSlugification(unittest.TestCase):
         self.assertEqual(r, '1000-reasons-you-are-1')
 
 
+class TestUtils(unittest.TestCase):
+
+    def test_smart_truncate_no_max_length(self):
+        txt = '1,000 reasons you are #1'
+        r = smart_truncate(txt)
+        self.assertEqual(r, txt)
+
+    def test_smart_truncate_no_seperator(self):
+        txt = '1,000 reasons you are #1'
+        r = smart_truncate(txt, max_length=100, separator='_')
+        self.assertEqual(r, txt)
+
 if __name__ == '__main__':
     unittest.main()