[Python-checkins] r54688 - python/trunk/Lib/test/test_softspace.py

[collin.winter]

Author: collin.winter
Date: Wed Apr  4 20:36:30 2007
New Revision: 54688

Modified:
   python/trunk/Lib/test/test_softspace.py
Log:
Make test_softspace use unittest.

Modified: python/trunk/Lib/test/test_softspace.py
==============================================================================
--- python/trunk/Lib/test/test_softspace.py	(original)
+++ python/trunk/Lib/test/test_softspace.py	Wed Apr  4 20:36:30 2007
@@ -1,14 +1,23 @@
-from test import test_support
+from test.test_support import run_unittest
+import unittest
 import StringIO
 
-# SF bug 480215:  softspace confused in nested print
-f = StringIO.StringIO()
-class C:
-    def __str__(self):
-        print >> f, 'a'
-        return 'b'
+class SoftspaceTests(unittest.TestCase):
+    def test_bug_480215(self):
+        # SF bug 480215:  softspace confused in nested print
+        f = StringIO.StringIO()
+        class C:
+            def __str__(self):
+                print >> f, 'a'
+                return 'b'
 
-print >> f, C(), 'c ', 'd\t', 'e'
-print >> f, 'f', 'g'
-# In 2.2 & earlier, this printed ' a\nbc  d\te\nf g\n'
-test_support.vereq(f.getvalue(), 'a\nb c  d\te\nf g\n')
+        print >> f, C(), 'c ', 'd\t', 'e'
+        print >> f, 'f', 'g'
+        # In 2.2 & earlier, this printed ' a\nbc  d\te\nf g\n'
+        self.assertEqual(f.getvalue(), 'a\nb c  d\te\nf g\n')
+        
+def test_main():
+    run_unittest(SoftspaceTests)
+    
+if __name__ == '__main__':
+    test_main()