tests: Compat restore tests with python2 for testing basic tests in old platforms

AndrewBille · AndrewBille · commit e2a94a5be3c3 · 2021-10-26T14:03:00.000+07:00
diff --git a/tests/restore.py b/tests/restore.py
@@ -4,7 +4,11 @@
 import subprocess
 import sys
 from time import sleep
-from datetime import datetime, timedelta, timezone
+from datetime import datetime, timedelta
+try:
+    from datetime import timezone
+except ImportError:  # py2compat
+    from pytz import timezone
 import hashlib
 import shutil
 import json
