diff options
Diffstat (limited to 'moses/TranslationModel/ProbingPT/tests/vocabid_test.cpp')
-rw-r--r-- | moses/TranslationModel/ProbingPT/tests/vocabid_test.cpp | 81 |
1 files changed, 41 insertions, 40 deletions
diff --git a/moses/TranslationModel/ProbingPT/tests/vocabid_test.cpp b/moses/TranslationModel/ProbingPT/tests/vocabid_test.cpp index bc82db74e..fac439eeb 100644 --- a/moses/TranslationModel/ProbingPT/tests/vocabid_test.cpp +++ b/moses/TranslationModel/ProbingPT/tests/vocabid_test.cpp @@ -3,43 +3,44 @@ #include "hash.hh" #include "vocabid.hh" -int main(int argc, char* argv[]){ - - //Create a map and serialize it - std::map<uint64_t, std::string> vocabids; - StringPiece demotext = StringPiece("Demo text with 3 elements"); - add_to_map(&vocabids, demotext); - //Serialize map - serialize_map(&vocabids, "/tmp/testmap.bin"); - - //Read the map and test if the values are the same - std::map<uint64_t, std::string> newmap; - read_map(&newmap, "/tmp/testmap.bin"); - - //Used hashes - uint64_t num1 = getHash(StringPiece("Demo")); - uint64_t num2 = getVocabID("text"); - uint64_t num3 = getHash(StringPiece("with")); - uint64_t num4 = getVocabID("3"); - uint64_t num5 = getHash(StringPiece("elements")); - uint64_t num6 = 0; - - //Tests - bool test1 = getStringFromID(&newmap, num1) == getStringFromID(&vocabids, num1); - bool test2 = getStringFromID(&newmap, num2) == getStringFromID(&vocabids, num2); - bool test3 = getStringFromID(&newmap, num3) == getStringFromID(&vocabids, num3); - bool test4 = getStringFromID(&newmap, num4) == getStringFromID(&vocabids, num4); - bool test5 = getStringFromID(&newmap, num5) == getStringFromID(&vocabids, num5); - bool test6 = getStringFromID(&newmap, num6) == getStringFromID(&vocabids, num6); - - - if (test1 && test2 && test3 && test4 && test5 && test6){ - std::cout << "Map was successfully written and read!" << std::endl; - } else { - std::cout << "Error! " << test1 << " " << test2 << " " << test3 << " " << test4 << " " << test5 << " " << test6 << std::endl; - } - - - return 1; - -} +int main(int argc, char* argv[]) +{ + + //Create a map and serialize it + std::map<uint64_t, std::string> vocabids; + StringPiece demotext = StringPiece("Demo text with 3 elements"); + add_to_map(&vocabids, demotext); + //Serialize map + serialize_map(&vocabids, "/tmp/testmap.bin"); + + //Read the map and test if the values are the same + std::map<uint64_t, std::string> newmap; + read_map(&newmap, "/tmp/testmap.bin"); + + //Used hashes + uint64_t num1 = getHash(StringPiece("Demo")); + uint64_t num2 = getVocabID("text"); + uint64_t num3 = getHash(StringPiece("with")); + uint64_t num4 = getVocabID("3"); + uint64_t num5 = getHash(StringPiece("elements")); + uint64_t num6 = 0; + + //Tests + bool test1 = getStringFromID(&newmap, num1) == getStringFromID(&vocabids, num1); + bool test2 = getStringFromID(&newmap, num2) == getStringFromID(&vocabids, num2); + bool test3 = getStringFromID(&newmap, num3) == getStringFromID(&vocabids, num3); + bool test4 = getStringFromID(&newmap, num4) == getStringFromID(&vocabids, num4); + bool test5 = getStringFromID(&newmap, num5) == getStringFromID(&vocabids, num5); + bool test6 = getStringFromID(&newmap, num6) == getStringFromID(&vocabids, num6); + + + if (test1 && test2 && test3 && test4 && test5 && test6) { + std::cout << "Map was successfully written and read!" << std::endl; + } else { + std::cout << "Error! " << test1 << " " << test2 << " " << test3 << " " << test4 << " " << test5 << " " << test6 << std::endl; + } + + + return 1; + +} |