googleapis
diff --git a/‎packages/google-cloud-dlp/tests/unit/gapic/dlp_v2/test_dlp_service.py‎
Lines changed: 9 additions & 2 deletions b/‎packages/google-cloud-dlp/tests/unit/gapic/dlp_v2/test_dlp_service.py‎
Lines changed: 9 additions & 2 deletions
@@ -13,9 +13,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-import math
 import os
 
+# try/except added for compatibility with python < 3.8
+try:
+    from unittest import mock
+    from unittest.mock import AsyncMock
+except ImportError:
+    import mock
+
+import math
+
 from google.api_core import gapic_v1, grpc_helpers, grpc_helpers_async, path_template
 from google.api_core import client_options
 from google.api_core import exceptions as core_exceptions
@@ -33,7 +41,6 @@
 from google.type import timeofday_pb2  # type: ignore
 import grpc
 from grpc.experimental import aio
-import mock
 from proto.marshal.rules.dates import DurationRule, TimestampRule
 import pytest