summaryrefslogtreecommitdiff
path: root/lib/test/test_shared.py
blob: fa7d24939808259a4d7a9e1b7c3955066d90efa0 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
import os

def resolve_imagename(ROOT_DIR, base_name):
    """Resolve image name for tests."""

    image_name = os.path.join(ROOT_DIR, base_name)

    # Is it zipped?
    if not os.path.isfile(image_name):
        image_name.replace('.fits', '.fits.gz')

    return image_name