Fix Python 2.6 compatibility in test

nbessi · Sep 30, 2014 · f56fa3c · f56fa3c
1 parent bdc1f9e
commit f56fa3c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyhiccup/tests/test_basic.py b/pyhiccup/tests/test_basic.py
@@ -18,6 +18,7 @@
 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 #
 ##############################################################################
+import sys
 import re
 try:
     import unittest2 as unittest
@@ -106,8 +107,7 @@ def test_xhtml_transitional_doc_type(self):
 
     def test_wrong_type(self):
         data = []
-        with self.assertRaises(ValueError):
-            html(data, etype='kaboom')
+        self.assertRaises(ValueError, html, data, 'kaboom')
 
 
 class XMLTest(CommonTest):