diff options
Diffstat (limited to 'awxkit/test/test_ws.py')
| -rw-r--r-- | awxkit/test/test_ws.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/awxkit/test/test_ws.py b/awxkit/test/test_ws.py index afc6b42fc5..ea55ba8ffa 100644 --- a/awxkit/test/test_ws.py +++ b/awxkit/test/test_ws.py @@ -1,7 +1,10 @@ # -*- coding: utf-8 -*- from collections import namedtuple -from unittest.mock import patch +try: + from unittest.mock import patch +except ImportError: + from mock import patch import pytest from awxkit.ws import WSClient |