NanobindTraitCheckStream.cpp

Go to the documentation of this file.

/***************************************
        Auteur : Thibaut Oprinsen
        Mail : thibaut.oprinsen@lapp.in2p3.fr
        Licence : CeCILL-C
****************************************/
 
#include "NanobindTraitCheckStream.h"
#include "type_utils.h"

NanobindTraitCheckStream::NanobindTraitCheckStream()
: PAbstractNanobindTraitBackend()
{}

NanobindTraitCheckStream::~NanobindTraitCheckStream(){}

void NanobindTraitCheckStream::registerStaticMethod(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode) const{
        if(!mode.enableCheckStream){return;}
        PString name(classConfig.getName());
        if(classConfig.getIsEnum()){
                name = name + "::" + name;
        }
        fs << std::endl << std::endl << "\t\t///Check stream method to compare two " << name << " objects";
        fs << std::endl << "\t\t.def_static(\"check_stream\", [](const std::string & fieldDescription, const " << name << " & data, const " << name << " & reference) -> bool {";
        fs << std::endl << "\t\t\tstd::stringstream out;";
        fs << std::endl << "\t\t\tif(!CheckStream<" << name << ">::check_stream(fieldDescription, data, reference, out)){";
        fs << std::endl << "\t\t\t\tthrow nb::value_error(out.str().c_str());";
        fs << std::endl << "\t\t\t}";
        fs << std::endl << "\t\t\treturn true;";
        fs << std::endl << "\t\t}, ";
        fs << std::endl << "\t\tnb::arg(\"fieldDescription\"), nb::arg(\"data\"), nb::arg(\"reference\"),";
        fs << std::endl << "\t\t\"Check stream for " << name << " (equivalent to CheckStream<" << name << ">::check_stream)\")";
}

void NanobindTraitCheckStream::testFunction(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode, const PString & baseFileName) const{
        if(!mode.enableUnitTest || !mode.enableCheckStream){return;}
        PString name(classConfig.getName());
        PString moduleName = baseFileName.toLower();
 
        fs << "def test_check_stream():" << std::endl;
        fs << "\t\"\"\"Test CheckStream functionality\"\"\"" << std::endl;
        fs << std::endl;
        fs << "\t# Create two identical " << name << std::endl;
        const PVecClassAttribute & listAttr(classConfig.getListAttribute());
        if(classConfig.getIsEnum()){
                if(listAttr.size() != 0lu){
                        fs << "\ttest_value1 = " << moduleName << "." << name << "." << listAttr.front().getName() << std::endl;
                        fs << "\ttest_value2 = " << moduleName << "." << name << "." << listAttr.front().getName() << std::endl;
                }
                fs << "\t# Test identical " << name << " - should return True" << std::endl;
                fs << "\tresult_identical = " << moduleName << "." << name << ".check_stream(\"Test identical\", test_value1, test_value2)" << std::endl;
                fs << "\tassert result_identical == True, \"CheckStream should return True for identical " << name << "\"" << std::endl;
        }else{
                fs << "\ttest_value1 = " << moduleName << "." << name << "()" << std::endl;
        
                for(PVecClassAttribute::const_iterator it(listAttr.begin()); it != listAttr.end(); ++it){
                        PString defaultValue = getTestDefaultValueTypeInPython(it->getType());
                        if(defaultValue != ""){
                                fs << "\ttest_value1." << it->getName() << " = " << defaultValue << std::endl;
                        }
                }
                fs << std::endl;
                fs << "\ttest_value2 = " << moduleName << "." << name << "()" << std::endl;
                for(PVecClassAttribute::const_iterator it(listAttr.begin()); it != listAttr.end(); ++it){
                        PString defaultValue = getTestDefaultValueTypeInPython(it->getType());
                        if(defaultValue != ""){
                                fs << "\ttest_value2." << it->getName() << " = " << defaultValue << std::endl;
                        }
                }
                fs << std::endl;
                fs << "\t# Test identical " << name << " - should return True" << std::endl;
                fs << "\tresult_identical = " << moduleName << "." << name << ".check_stream(\"Test identical\", test_value1, test_value2)" << std::endl;
                fs << "\tassert result_identical == True, \"CheckStream should return True for identical " << name << "\"" << std::endl;
                fs << std::endl;
        }
}

void NanobindTraitCheckStream::testCallFunction(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode) const{
        if(!mode.enableUnitTest || !mode.enableCheckStream){return;}
        PString name(classConfig.getName());
        fs << "test_check_stream()" << std::endl;
}