diff options
Diffstat (limited to 'cmdtestlib.py')
-rw-r--r-- | cmdtestlib.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/cmdtestlib.py b/cmdtestlib.py index a42b459..a064cbc 100644 --- a/cmdtestlib.py +++ b/cmdtestlib.py @@ -46,8 +46,9 @@ class TestDir(object): script_names = ['setup_once', 'setup', 'teardown', 'teardown_once'] for name in script_names: - if name in filenames: - setattr(self, name, os.path.join(dirname, name)) + full_path = os.path.join(dirname, name) + if os.path.exists(full_path): + setattr(self, name, full_path) prefixes = self.find_prefixes(filenames) for prefix in prefixes: |