test_recognize.cpp 1.33 KB
/* 
 * File:   test_recognize.cpp
 * Author: mlenart
 *
 * Created on October 30, 2013, 5:26 PM
 */

#include <cstdlib>
#include <sstream>
#include "fsa.hpp"
#include "utils.hpp"

using namespace std;

void doTest(const FSA<char*>& fsa, const char* fname) {
    ifstream ifs;
    //    ifs.exceptions(std::ifstream::failbit | std::ifstream::badbit);
    ifs.open(fname, ios::binary);
    string line;
    while (getline(ifs, line)) {
#ifdef DEBUG_BUILD
        cerr << "RECOGNIZE " << line << endl;
#endif
        vector<string> splitVector(split(line, '\t'));
        string key = splitVector[0];

        //        cerr << "test " << key << endl;

        char* value2;
        validate(fsa.tryToRecognize(key.c_str(), value2), "Failed to recognize " + key);
    }
    //    validate(ifs.good(), "Something wrong with the input file");
    //    validate(!ifs.fail(), "Something wrong with the input file");
    validate(ifs.eof(), "Failed to read the input file to the end");
}

int main(int argc, char** argv) {
    validate(argc == 3, "Must provide exactly two arguments - FSA filename and dictionary filename.");
    const unsigned char* fsaData = readFile(argv[1]);
    StringDeserializer deserializer;
    FSA<char*>* fsa = FSA<char*>::getFSA(fsaData, deserializer);
    doTest(*fsa, argv[2]);
    //    cout << argc << endl;
    delete fsa;
    return 0;
}