diff options
Diffstat (limited to 'tests/integration/models_services_test.py')
| -rw-r--r-- | tests/integration/models_services_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/integration/models_services_test.py b/tests/integration/models_services_test.py index 827242a..39cccf8 100644 --- a/tests/integration/models_services_test.py +++ b/tests/integration/models_services_test.py @@ -6,6 +6,7 @@ from .. import helpers from .base import TEST_API_VERSION from docker.errors import InvalidArgument from docker.types.services import ServiceMode +import pytest class ServiceTest(unittest.TestCase): @@ -265,8 +266,7 @@ class ServiceTest(unittest.TestCase): while len(tasks) == 0: tasks = service.tasks() assert len(tasks) == 1 - with self.assertRaises(InvalidArgument, - msg='Cannot scale a global container'): + with pytest.raises(InvalidArgument): service.scale(2) assert len(tasks) == 1 |