From 96e853995203aae902416b4ba82707161ac6170c 2011-12-08 21:49:52
From: Thomas Kluyver <takowl@gmail.com>
Date: 2011-12-08 21:49:52
Subject: [PATCH] Add tests for prompt system.

---

diff --git a/IPython/core/tests/test_prompts.py b/IPython/core/tests/test_prompts.py
new file mode 100644
index 0000000..d3613b3
--- /dev/null
+++ b/IPython/core/tests/test_prompts.py
@@ -0,0 +1,43 @@
+"""Tests for prompt generation."""
+
+import unittest
+
+import nose.tools as nt
+
+from IPython.testing import tools as tt, decorators as dec
+from IPython.core.prompts import PromptManager
+from IPython.testing.globalipapp import get_ipython
+
+ip = get_ipython()
+
+
+class PromptTests(unittest.TestCase):
+    def setUp(self):
+        self.pm = PromptManager(shell=ip, config=ip.config)
+    
+    def test_multiline_prompt(self):
+        self.pm.in_template = "[In]\n>>>"
+        self.pm.render('in')
+        self.assertEqual(self.pm.width, 3)
+        self.assertEqual(self.pm.txtwidth, 3)
+        
+        self.pm.in_template = '[In]\n'
+        self.pm.render('in')
+        self.assertEqual(self.pm.width, 0)
+        self.assertEqual(self.pm.txtwidth, 0)
+    
+    def test_translate_abbreviations(self):
+        def do_translate(template):
+            self.pm.in_template = template
+            return self.pm.templates['in']
+        
+        pairs = [(r'%n>', '{color.number}{count}{color.prompt}>'),
+                 (r'\T', '{time}'),
+                 (r'\n', '\n')
+                ]
+    
+        tt.check_pairs(do_translate, pairs)
+    
+    def test_render(self):
+        self.pm.in_template = r'\#>'
+        self.assertEqual(self.pm.render('in',color=False), '%d>' % ip.execution_count)