diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/google/protobuf/internal/well_known_types_test.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/python/google/protobuf/internal/well_known_types_test.py b/python/google/protobuf/internal/well_known_types_test.py index 60b0c47d..0e31e6f8 100644 --- a/python/google/protobuf/internal/well_known_types_test.py +++ b/python/google/protobuf/internal/well_known_types_test.py @@ -36,10 +36,14 @@ __author__ = 'jieluo@google.com (Jie Luo)' from datetime import datetime +try: + import unittest2 as unittest +except ImportError: + import unittest + from google.protobuf import duration_pb2 from google.protobuf import field_mask_pb2 from google.protobuf import timestamp_pb2 -import unittest from google.protobuf import unittest_pb2 from google.protobuf.internal import test_util from google.protobuf.internal import well_known_types |