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());
        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() + "_module";
 
        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;
        fs << "\ttest_value1 = " << moduleName << "." << name << "()" << std::endl;
        const PVecClassAttribute & listAttr(classConfig.getListAttribute());
        for(PVecClassAttribute::const_iterator it(listAttr.begin()); it != listAttr.end(); ++it){
                PString defaultValue = getTestDefaultValueTypeInPython(it->getType());
                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());
                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;
        fs << "\t# Create a different " << name << std::endl;
        fs << "\ttest_value3 = " << moduleName << "." << name << "()" << std::endl;
        for(PVecClassAttribute::const_iterator it(listAttr.begin()); it != listAttr.end(); ++it){
                PString defaultValue = getTestDefaultValueTypeInPython(it->getType());
                PString differentValue = defaultValue;
        
                if(it == listAttr.begin()){
                        if(it->getType() == "int" || it->getType() == "short" || it->getType() == "long" || it->getType() == "size_t"){
                                differentValue = defaultValue + " + 1";  // 42 + 1 = 43
                        }else if(it->getType() == "float" || it->getType() == "double"){
                                differentValue = defaultValue + " + 0.5";  // 1.0 + 0.5 = 1.5
                        }else if(it->getType() == "bool"){
                                differentValue = "not " + defaultValue;  // not True = False
                        }else if(it->getType() == "std::string" || it->getType() == "PString"){
                                differentValue = defaultValue + " + \"_modified\"";  // "Some string" + "_modified"
                        }else{
                                differentValue = "\"different_value\"";
                        }
                }
                
                fs << "\ttest_value3." << it->getName() << " = " << differentValue << std::endl;
        }
        fs << std::endl;
        fs << "\t#test different " << name << " - should return False" << std::endl;
        fs << "\ttry:" << std::endl;
        fs << "\t\tresult_different = " << moduleName << "." << name << ".check_stream(\"Test different\", test_value1, test_value3)" << std::endl;
        fs << "\t\tassert False, \"CheckStream should raise an exception for different " << name << "\"" << std::endl << std::endl;
        fs << "\texcept:" << std::endl;
        fs << "\t\tassert True" << std::endl;
        // fs << "\tassert result_different == False, \"CheckStream should return False for different " << name << "\"" << std::endl << 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;
}