Fix py2 tests

holoviz · May 14, 2021 · 25f35ba · 25f35ba
1 parent 6d360b0
commit 25f35ba
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/tests/API0/teststringparam.py b/tests/API0/teststringparam.py
@@ -23,7 +23,8 @@ class A(param.Parameterized):
 
         a = A()
 
-        exception = "String parameter 's' only takes a string value, not value of type <class 'NoneType'>."
+        cls = 'class' if sys.version_info.major > 2 else 'type'
+        exception = "String parameter 's' only takes a string value, not value of type <%s 'NoneType'>." % cls
         with self.assertRaisesRegexp(ValueError, exception):
             a.s = None  # because allow_None should be False
 

diff --git a/tests/API1/teststringparam.py b/tests/API1/teststringparam.py
@@ -1,6 +1,8 @@
 """
 Unit test for String parameters
 """
+import sys
+
 from . import API1TestCase
 
 import param
@@ -23,7 +25,8 @@ class A(param.Parameterized):
 
         a = A()
 
-        exception = "String parameter 's' only takes a string value, not value of type <class 'NoneType'>."
+        cls = 'class' if sys.version_info.major > 2 else 'type'
+        exception = "String parameter 's' only takes a string value, not value of type <%s 'NoneType'>." % cls
         with self.assertRaisesRegexp(ValueError, exception):
             a.s = None  # because allow_None should be False