diff --git a/src/testing/common/database.py b/src/testing/common/database.py index 5ba8c5e..74e1067 100644 --- a/src/testing/common/database.py +++ b/src/testing/common/database.py @@ -23,7 +23,10 @@ from time import sleep from shutil import copytree, rmtree from datetime import datetime -import collections +try: + from collections.abc import Callable +except ImportError: + from collections import Callable class DatabaseFactory(object): @@ -285,7 +288,7 @@ def decorator(fn, path=arg): return skipIf(cond, "%s not found" % self.name)(fn) - if isinstance(arg, collections.Callable): # execute as simple decorator + if isinstance(arg, Callable): # execute as simple decorator return decorator(arg, None) else: # execute with path argument return decorator