Add multi type of char blocks width test

2 files changed, 29 insertions, 0 deletions

diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/__init__.py
diff --git a/tests/test_utils.py b/tests/test_utils.py
new file mode 100644
index 0000000..6ed710b
--- /dev/null
+++ b/tests/test_utils.py
@@ -0,0 +1,29 @@
+# -*- coding:utf-8 -*-
+from __future__ import unicode_literals
+import unittest
+
+from prettytable.prettytable import _char_block_width
+
+
+
+def _width_test_factory(width, words):
+    def _(self):
+        map(lambda x: self.assertEqual(width, _char_block_width(ord(x))),
+            list(words))
+    return _
+
+
+class CharBlockWidthTest(unittest.TestCase):
+    fixtures = {
+        'normal': (1, '12345qwerljk/.,WESD'),
+        'chs': (2, '石室诗士施氏嗜狮誓食十狮'),
+        'jp': (2, 'はじめまして'),
+        'hangul': (2, '우리글자언문청'),
+        'full_width_latin': (2, 'ＸＹＺ［＼］＾＿ｘｙｚ｛｜｝～｟'),
+        # 'cjk': (2, u''),
+    }
+
+    def test_fixtures(self):
+        for name in self.fixtures:
+            _width_test_factory(*self.fixtures[name])(self)
+