TestCommon.cc

#include "TestCommon.h"

//----------------------------------------------------------------
// Method to check whether two DataFrames are equal
// @param data1:    first DataFrame to check
// @param data2:    second DataFrame to check
// @param fatalErr: set to nonzero if dimensions are not the same
// @return a list of the rows of the dataframe that were different
//----------------------------------------------------------------
std::vector< int > CheckDataFrameEquality ( DataFrame< double > data1, 
                                            DataFrame< double > data2,
                                            int & fatalErr ) {
    // to keep track of the rows with different content
    std::vector< int > badRows;

    // to hold each row for comparison
    std::valarray< double > row1;
    std::valarray< double > row2;

    // check if dimensions are equal
    if ( data1.NRows()    != data2.NRows() or
         data1.NColumns() != data2.NColumns() ) {
        fatalErr = 1;
    }

    // check if any rows were different in the dataframes passed in
    for ( size_t rowIdx = 0; rowIdx < data1.NRows(); rowIdx++ ) {
        for ( size_t colIdx = 0; colIdx < data1.NColumns(); colIdx++ ) {

            float diff = std::abs(data1(rowIdx,colIdx) - data2(rowIdx,colIdx));

            if ( diff >= EPSILON ) {
                badRows.push_back( rowIdx );
                break;
            }
        }    
    }

    return badRows;
}

//----------------------------------------------------------------
// Run DataFrame comparison
// @param testName: description of the test being run
// @param data1:    first DataFrame to check
// @param data2:    second DataFrame to check
// @return:         none, just prints to stdout
//----------------------------------------------------------------
void MakeTest (std::string testName, DataFrame< double > data1, 
                                     DataFrame< double > data2 ) {

    int fatalErr = 0;

    //the rows that were different in the test
    std::vector< int > badRows = CheckDataFrameEquality(data1, data2, fatalErr);

    if ( fatalErr ) {
        std::cout << RED_TEXT << "FATAL ERR: dimensions of block 1: "
                  << data1.NRows() << "x" << data1.NColumns()
                  << " do not equal dimensions of block 2: "
                  << data2.NRows() << "x " << data2.NColumns() << std::endl;
    }

    // print info on test name and test results
    std::cout << STR_LINE_SEP << std::endl;
    std::cout << "Test: " << testName << std::endl;
    std::cout << STR_LINE_SEP << std::endl;

    if ( badRows.empty() ) {
        std::cout << GREEN_TEXT; 
        std::cout << TAB_CHAR << "PASSED. EPSILON: " << EPSILON;
    }
    else {
        int numBadRows =
            std::count_if( badRows.begin(), 
                           badRows.end(), [](int i){ return i != 0; } );

        if ( numBadRows ) {
            std::cout << YELLOW_TEXT; 
            std::cout << TAB_CHAR << "FAILED. EPSILON: " << EPSILON;
        }

#ifdef PRINT_DIFFERENCE_IN_RESULTS
        std::cout << TAB_CHAR << TAB_CHAR << "Block 1 column names: ";
        for (auto colName : data1.ColumnNames()) { 
            std::cout << std::setw(10) << colName << TAB_CHAR;
        }
        std::cout << std::endl;
        std::cout << TAB_CHAR << TAB_CHAR << "Block 2 column names: ";
        for (auto colName : data2.ColumnNames()) { 
            std::cout << std::setw(10) << colName << TAB_CHAR;
        }
        std::cout << std::endl;

        std::cout << TAB_CHAR << "first 10 different rows:" << std::endl;
        
        for (auto iterate  = badRows.begin();
                  iterate != badRows.end(); 
                ++iterate) {

            std::valarray< double > badRow1 = data1.Row( *iterate );
            std::valarray< double > badRow2 = data2.Row( *iterate );

            std::cout << TAB_CHAR << TAB_CHAR << "Block 1 row "<<*iterate<<": ";
            for (double elem : badRow1) {
                std::cout << std::setw(10) << TAB_CHAR << elem << " ";
            }
            std::cout << std::endl;
            std::cout << TAB_CHAR << TAB_CHAR << "Block 2 row "<<*iterate<<": ";
            for (double elem : badRow2) {
                std::cout << std::setw(10) << TAB_CHAR << elem << " ";
            }
            std::cout << std::endl << std::endl;
        }

#endif
    }
    
    std::cout << RESET_TEXT << std::endl << std::flush;
}