Commit 7237395c authored by Jan Kiene's avatar Jan Kiene
Browse files

use new script in tests

parent eddac253

tests/constants.py

+1 −3
Original line number Diff line number Diff line
@@ -219,7 +219,7 @@ INPUT_EXPERIMENT_NAMES = [ 
    "BS1534-7a",

    "BS1534-7b",

    "P800-1",

    # P800-2 is tested category-wise, see below

    "P800-2",

    "P800-3",

    "P800-4",

    "P800-5",
@@ -230,5 +230,3 @@ INPUT_EXPERIMENT_NAMES = [ 
]

__LABS = cycle("abcd")

LAB_IDS_FOR_EXPERIMENTS = [next(__LABS) for e in INPUT_EXPERIMENT_NAMES]
 


CREATE_CATEGORIES_TESTS = ["P800-2"]

tests/test_experiments.py

+7 −34
Original line number Diff line number Diff line
@@ -34,6 +34,7 @@ import shutil 
from pathlib import Path



import pytest

import sys

from numpy.random import random, seed



from ivas_processing_scripts import main as generate_test
@@ -41,23 +42,15 @@ from ivas_processing_scripts.audiotools import audio 
from ivas_processing_scripts.audiotools.audiofile import concat, write

from ivas_processing_scripts.processing.config import TestConfig

from tests.constants import (

    CREATE_CATEGORIES_TESTS,

    EXPERIMENTS_DIR,

    FORMAT_TO_METADATA_FILES,

    INPUT_EXPERIMENT_NAMES,

    NCHAN_TO_FILE,

    LAB_IDS_FOR_EXPERIMENTS,

    TESTS_DIR,

)

from experiments import create_experiment_config, create_items



BG_NOISE_NAME = "background_noise_cat.wav"





class Arguments:

    def __init__(self, config):

        self.config = config

        self.debug = True

HERE = Path(__file__).parent.absolute()

sys.path.append(HERE.parent)

from generate_test import create_experiment_setup, Arguments





def setup_input_files_for_config(config):
@@ -108,30 +101,10 @@ def setup_input_files_for_config(config): 
@pytest.mark.parametrize("exp_lab_pair", zip(INPUT_EXPERIMENT_NAMES, LAB_IDS_FOR_EXPERIMENTS))

def test_generate_test_items(exp_lab_pair):

    exp_name, lab_id = exp_lab_pair

    cfgs = sorted(create_experiment_config.create_experiment_setup(exp_name, lab_id))

    cfg = cfgs[0]



    args = Arguments(cfg)

    config = TestConfig(cfg)

    setup_input_files_for_config(config)

    generate_test(args)

    cfgs = create_experiment_setup(exp_name, lab_id)





@pytest.mark.parametrize("exp_name", CREATE_CATEGORIES_TESTS)

@pytest.mark.parametrize("lab_id", ["a", "b", "c", "d"])

def test_categories(exp_name, lab_id):

    cfgs = create_experiment_config.create_experiment_setup(exp_name, lab_id)

    for cfg in cfgs:

        args = Arguments(str(cfg))

        config = TestConfig(cfg)

        setup_input_files_for_config(config)



    create_items.create_items(exp_name, lab_id)





@pytest.mark.parametrize("exp_name", [f"P800-{i}" for i in range(1, 10)])

@pytest.mark.parametrize("lab_id", ["a", "b", "c", "d"])

def test_experiment_setup(exp_name, lab_id):

    cfgs = create_experiment_config.create_experiment_setup(exp_name, lab_id)

    for cfg in cfgs:

        # create config for validation of folder structure to be present

        _ = TestConfig(cfg)
 
        generate_test(args)