aboutsummaryrefslogtreecommitdiffhomepage
path: root/python/google/protobuf/internal/wire_format_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/google/protobuf/internal/wire_format_test.py')
-rwxr-xr-xpython/google/protobuf/internal/wire_format_test.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/python/google/protobuf/internal/wire_format_test.py b/python/google/protobuf/internal/wire_format_test.py
index 78dc1167..f659d18e 100755
--- a/python/google/protobuf/internal/wire_format_test.py
+++ b/python/google/protobuf/internal/wire_format_test.py
@@ -34,7 +34,10 @@
__author__ = 'robinson@google.com (Will Robinson)'
-import unittest
+try:
+ import unittest2 as unittest
+except ImportError:
+ import unittest
from google.protobuf import message
from google.protobuf.internal import wire_format