Commit f4803c57 authored by Ambra Di Piano's avatar Ambra Di Piano
Browse files

map template

parent c091ad23

astrort/testing/test_utils/test_utils.py

+3 −0
Original line number Diff line number Diff line
@@ -91,3 +91,6 @@ def test_get_all_seeds(test_conf_file): 
    conf = load_yaml_conf(test_conf_file)

    seeds = get_all_seeds(conf['simulator'])

    assert seeds.all() == np.array([1, 2]).all()



def test_map_template():

    assert map_template() == join(dirname(abspath(__file__)).replace('testing/test_utils', 'templates'), 'base_empty_map.fits')
 
 No newline at end of file

astrort/utils/utils.py

+8 −2
Original line number Diff line number Diff line
@@ -9,9 +9,12 @@ 
import random

import numpy as np

from os import listdir

from os.path import join, expandvars

from os.path import join, expandvars, dirname, abspath



def seeds_to_string_formatter_files(samples, output, name, seed, ext):

def map_template():

    return join(dirname(abspath(__file__)).replace('utils', 'templates'), 'base_empty_map.fits')



def seeds_to_string_formatter_files(samples, output, name, seed, ext, suffix=None):

    if samples <= 1e3:

        name = join(output, f"{name}_{seed:03d}.{ext}")

    elif samples <= 1e5:
@@ -20,6 +23,9 @@ def seeds_to_string_formatter_files(samples, output, name, seed, ext): 
        name = join(output, f"{name}_{seed:08d}.{ext}")

    else:

        name = join(output, f"{name}_{seed}.{ext}")

    # suffix

    if suffix is not None:

        name.replace(f'.{ext}', f'_{suffix}.{ext}')

    return name



def seeds_to_string_formatter(samples, name, seed):