Source code for pecos.io.write_test_results

import numpy as np
import logging

try:
    from nose.tools import nottest
except ImportError:
    def nottest(afunction):
        return afunction

logger = logging.getLogger(__name__)

@nottest
def write_test_results(filename, test_results):
    """
    Write test results file

    Parameters
    -----------
    filename : string
        Filename with full path

    test_results : pd.DataFrame
        Test results stored in pm.test_results
    """

    test_results.sort_values(['System Name', 'Variable Name'], inplace=True)
    test_results.index = np.arange(1, test_results.shape[0]+1)

    logger.info("Writing test results csv file " + filename)

    fout = open(filename, 'w')
    test_results.to_csv(fout, na_rep = 'NaN')
    fout.close()