"""
test one instance of SimDeep
"""

from os.path import abspath
from os.path import split

from os.path import isfile
from os.path import isdir

from os import remove
from shutil import rmtree



def test_instance():
    """
    test one instance of SimDeep
    """
    from simdeep.simdeep_analysis import SimDeep
    from simdeep.extract_data import LoadData

    PATH_DATA = '{0}/../examples/data/'.format(split(abspath(__file__))[0])

    TRAINING_TSV = {'RNA': 'rna_dummy.tsv', 'METH': 'meth_dummy.tsv', 'MIR': 'mir_dummy.tsv'}
    SURVIVAL_TSV = 'survival_dummy.tsv'

    PROJECT_NAME = 'TestProject'
    EPOCHS = 3

    dataset = LoadData(path_data=PATH_DATA,
                   survival_tsv=SURVIVAL_TSV,
                   training_tsv=TRAINING_TSV)

    deep_model_additional_args = {
        "epochs":EPOCHS, "seed":4}

    simdeep = SimDeep(dataset=dataset,
                      project_name=PROJECT_NAME,
                      path_results=PATH_DATA,
                      deep_model_additional_args=deep_model_additional_args,
                      )
    simdeep.load_training_dataset()
    simdeep.fit()
    simdeep.predict_labels_on_full_dataset()
    simdeep.predict_labels_on_test_fold()

    simdeep.load_new_test_dataset(
        {'RNA': 'rna_test_dummy.tsv'},
        'survival_test_dummy.tsv',
        'dummy')

    simdeep.predict_labels_on_test_dataset()

    from glob import glob

    for fil in glob('{0}/{1}*'.format(PATH_DATA, PROJECT_NAME)):
        if isfile(fil):
            remove(fil)
        elif isdir(fil):
            rmtree(fil)


if __name__ == '__main__':
    test_instance()