diff --git a/ddt.py b/ddt.py index 030c76d..8e14ae1 100644 --- a/ddt.py +++ b/ddt.py @@ -13,7 +13,7 @@ from functools import wraps try: import yaml -except Exception: +except ImportError: # pragma: no cover _have_yaml = False else: _have_yaml = True diff --git a/test/test_example.py b/test/test_example.py index 79bd2c7..dbf29bd 100644 --- a/test/test_example.py +++ b/test/test_example.py @@ -4,7 +4,7 @@ from test.mycode import larger_than_two, has_three_elements, is_a_greeting try: import yaml -except Exception: +except ImportError: # pragma: no cover have_yaml_support = False else: have_yaml_support = True