Merge ef76981 into 146cfdb

ValvePython · Feb 23, 2021 · 0e21814 · 0e21814
2 parents 146cfdb + ef76981
commit 0e21814
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_vdf.py b/tests/test_vdf.py
@@ -1,7 +1,11 @@
 import unittest
-import mock
 import sys
 
+try:
+    from unittest import mock
+except ImportError:
+    import mock
+
 try:
         from StringIO import StringIO
 except ImportError: