Commit 356a9eca authored by Ambra Di Piano's avatar Ambra Di Piano
Browse files

get all seeds

parent 25ac5fec

astrort/testing/test_utils/test_utils.py

+8 −0
Original line number Diff line number Diff line
@@ -7,8 +7,10 @@ 
# *****************************************************************************



import pytest

import numpy as np

from os.path import isdir

from astrort.utils.utils import *

from astrort.utils.wrap import *



@pytest.mark.test_tmp_folder

@pytest.mark.parametrize('samples', [3, 5, 8, 10])
@@ -83,3 +85,9 @@ def test_select_irf(array): 
    assert array in irf.lower()

    assert 'share/caldb/data/cta' in join(path, irf).lower()

    assert isdir(join(path, irf)) is True



@pytest.mark.test_conf_file

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()

astrort/utils/utils.py

+8 −1
Original line number Diff line number Diff line
@@ -7,6 +7,7 @@ 
# *****************************************************************************



import random

import numpy as np

from os import listdir

from os.path import join, expandvars
@@ -69,3 +70,9 @@ def select_random_irf(array, prod): 
    irfs = listdir(path)

    irf = random.choice([i for i in irfs if array in i.lower()])

    return irf



def get_all_seeds(simulator):

    start_seed = simulator['seed']

    samples = simulator['samples']

    seeds = np.arange(start_seed, samples+start_seed, step=1)

    return seeds
 
 No newline at end of file