/home/palakars/workspace-inprogress/code-in-progress/aitools3/aitools3-aq-invertedindex-cpp/test/util/TestVocabulary.cpp

Go to the documentation of this file.

#include <boost/test/unit_test.hpp>
#include "Vocabulary.hpp"
#include "Converter.hpp"
#include <sstream>

using namespace aitools::invertedindex;

BOOST_AUTO_TEST_SUITE(vocabulary);

BOOST_AUTO_TEST_CASE(default_ctor)
{
        Vocabulary vocabulary;
        BOOST_CHECK(vocabulary.is_empty());
        BOOST_CHECK_EQUAL(vocabulary.size(), 0);
}

BOOST_AUTO_TEST_CASE(explicit_ctor)
{
        BOOST_CHECK_THROW(Vocabulary("/does/not/exist.txt"), std::invalid_argument);
        // the non-throwing instantiation is tested by 'save_load'
}

BOOST_AUTO_TEST_CASE(copy_ctor)
{
        Vocabulary vocabulary;
        vocabulary.insert("a");
        vocabulary.insert("b");
        vocabulary.insert("c");
        Vocabulary deep_copy(vocabulary);
        BOOST_CHECK_EQUAL(deep_copy.size(), vocabulary.size());
        BOOST_CHECK(deep_copy.contains("a"));
        BOOST_CHECK(deep_copy.contains("b"));
        BOOST_CHECK(deep_copy.contains("c"));
}

BOOST_AUTO_TEST_CASE(accumulate)
{
        Vocabulary vocabulary;
        vocabulary.insert("a", 10);
        vocabulary.accumulate("a", 10);
        BOOST_CHECK_EQUAL(vocabulary.find("a")->second, 20);
        vocabulary.accumulate("b", 10);
        vocabulary.accumulate("b", 10);
        BOOST_CHECK_EQUAL(vocabulary.find("b")->second, 20);
}

BOOST_AUTO_TEST_CASE(clear)
{
        Vocabulary vocabulary;
        vocabulary.insert("a");
        vocabulary.insert("b");
        vocabulary.insert("c");
        BOOST_CHECK_EQUAL(vocabulary.size(), 3);
        vocabulary.clear();
        BOOST_CHECK(vocabulary.is_empty());
}

BOOST_AUTO_TEST_CASE(contains)
{
        Vocabulary vocabulary;
        vocabulary.insert("a");
        vocabulary.insert("b");
        BOOST_CHECK(vocabulary.contains("a"));
        BOOST_CHECK(vocabulary.contains("b"));
        vocabulary.accumulate("c", 0);
        vocabulary.accumulate("d", 0);
        BOOST_CHECK(vocabulary.contains("c"));
        BOOST_CHECK(vocabulary.contains("d"));
}

BOOST_AUTO_TEST_CASE(find)
{
        Vocabulary vocabulary;
        vocabulary.insert("a", 10);
        BOOST_CHECK(vocabulary.find("a") != vocabulary.end());
        BOOST_CHECK(vocabulary.find("b") == vocabulary.end());
        BOOST_CHECK_EQUAL(vocabulary.find("a")->first, "a");
        BOOST_CHECK_EQUAL(vocabulary.find("a")->second, 10);
}

BOOST_AUTO_TEST_CASE(get_words)
{
        Vocabulary vocabulary;
        vocabulary.insert("a");
        vocabulary.insert("b");
        vocabulary.insert("c");
        std::vector<std::string> words;
        vocabulary.get_words(words);
        BOOST_CHECK_EQUAL(words.size(), vocabulary.size());
        for (unsigned i(0); i != words.size(); ++i)
        {
                BOOST_CHECK_EQUAL(words[i], vocabulary.find(words[i])->first);
        }
}

BOOST_AUTO_TEST_CASE(save_load)
{
        Vocabulary vocabulary;
        vocabulary.insert("a", 10);
        vocabulary.insert("b", 20);
        vocabulary.insert("c", 30);
        bfs::path file("aitools.invertedindex.vocabulary.txt");
        vocabulary.save(file);
        Vocabulary deep_copy(file);
        BOOST_CHECK_EQUAL(deep_copy.find("a")->first, "a");
        BOOST_CHECK_EQUAL(deep_copy.find("a")->second, 10);
        BOOST_CHECK_EQUAL(deep_copy.find("b")->first, "b");
        BOOST_CHECK_EQUAL(deep_copy.find("b")->second, 20);
        BOOST_CHECK_EQUAL(deep_copy.find("c")->first, "c");
        BOOST_CHECK_EQUAL(deep_copy.find("c")->second, 30);
        deep_copy.clear();
        deep_copy.load(file);
        BOOST_CHECK_EQUAL(deep_copy.find("a")->first, "a");
        BOOST_CHECK_EQUAL(deep_copy.find("a")->second, 10);
        BOOST_CHECK_EQUAL(deep_copy.find("b")->first, "b");
        BOOST_CHECK_EQUAL(deep_copy.find("b")->second, 20);
        BOOST_CHECK_EQUAL(deep_copy.find("c")->first, "c");
        BOOST_CHECK_EQUAL(deep_copy.find("c")->second, 30);
        bfs::remove(file);
}

BOOST_AUTO_TEST_SUITE_END();

---