move test to tests

thomasballinger · thomasballinger · commit 269bed754b79 · 2013-10-14T00:13:33.000-04:00
--HG--
branch : scroll-frontend
diff --git a/bpython/scrollfrontend/abbreviate.py b/bpython/scrollfrontend/abbreviate.py
@@ -38,9 +38,3 @@ def update_abbrs_with_vim_abbrs():
     abbrs = python_vim_abbreviations()
     ABBR.update(abbrs)
 
-if __name__ == '__main__':
-    abbrs = python_vim_abbreviations()
-    print abbrs
-    ABBR.update(abbrs)
-    print(substitute_abbreviations(0, 'improt asdf'))
-    print(substitute_abbreviations(0, 'foo(x, y() - 2.3242) + "asdf"'))
diff --git a/bpython/test/test_abbreviate.py b/bpython/test/test_abbreviate.py
@@ -0,0 +1,13 @@
+import bpython.scrollfrontend.abbreviate as abbreviate
+import unittest
+
+class TestAbbreviate(unittest.TestCase):
+
+    def test_substitution(self):
+        self.assertEqual(abbreviate.substitute_abbreviations(0, 'improt asdf'), (0, 'import asdf'))
+
+    def test_no_substitution(self):
+        self.assertEqual(abbreviate.substitute_abbreviations(0, 'foo(x, y() - 2.3242) + "asdf"'), (0, 'foo(x, y() - 2.3242) + "asdf"'))
+
+if __name__ == '__main__':
+    unittest.main()
