pw_test.h

Go to the documentation of this file.

#ifndef __PWTEST
#define __PWTEST
#include "gtest/gtest.h"
#include <iostream>
using namespace std;
extern int nproc_in_pool, rank_in_pool;
extern string precision_flag, device_flag;
 
class PWTEST: public testing::Test
{
public:
    static void SetUpTestCase()
    {
        if(rank_in_pool == 0)
        {
            cout<<"\033[32m"<<"============================"<<"\033[0m"<<endl;
            cout<<"\033[32m"<<"=    PW GPU MODULE TEST START  ="<<"\033[0m"<<endl;
            cout<<"\033[32m"<<"============================"<<"\033[0m"<<endl;
        }
    }
    static void TearDownTestCase()
    {
        if(rank_in_pool == 0)
        {
            cout<<"\033[32m"<<"============================"<<"\033[0m"<<endl;
            cout<<"\033[32m"<<"=     PW GPU MODULE TEST END   ="<<"\033[0m"<<endl;
            cout<<"\033[32m"<<"============================"<<"\033[0m"<<endl;
        }
    }
    void SetUp()
    {
        cout<<"\033[32m"<<"[    CASE  ]"<<"\033[0m"<<" ";
    }
    void TearDown(){}
};
 
#endif