trie_map_test.cpp

// pacakge
#include "trie.hpp"
#include "trie_map.hpp"
// c++
#include <vector>
#include <string>
#include <iostream>
// c
#include <cassert>

int main(const int argc, const char** argv) {
  using namespace std;
  using namespace com_masaers;
  
  vector<string> keys{ "aa", "a", "abb", "aaaa" };
  
  // TODO test erase_suffixes
  unordered_trie_map<char, size_t> uts;
  size_t count = 0;
  for (const auto& key : keys) {
    uts.insert(make_pair(key, count));
    ++count;
  }
  for (auto it = begin(uts); it != end(uts); ++it) {
    cout << '"';
    for (const auto& e : uts.path_to(it)) {
      cout << e;
    }
    cout << "\"\t" << it->second << endl;
  }
  {
    assert(uts.size() == keys.size());
    assert(uts.find(string("aa")) != uts.end());
    assert(uts.cfind(string("aa")) != uts.cend());
    uts.erase(string("aa"));
    assert(uts.size() == keys.size() - 1);
    assert(uts.find(string("aa")) == uts.end());
    assert(uts.cfind(string("aa")) == uts.cend());
    uts.clear();
    assert(uts.empty());
    assert(uts.size() == 0);
    assert(uts.find(string("a")) == uts.end());
    assert(uts.cfind(string("a")) == uts.cend());
  }
  
  ordered_trie_map<char, size_t> ots;
  for (const auto& key : keys) {
    ots.insert(make_pair(key, count));
    ++count;
  }
  for (auto it = begin(ots); it != end(ots); ++it) {
    cout << '"';
    for (const auto& e : ots.path_to(it)) {
      cout << e;
    }
    cout << "\"\t" << it->second << endl;
  }
  for (auto node : ots) {
    cout << '"';
    for (const auto& e : ots.path_to(node.first)) {
      cout << e;
    }
    cout << "\"\t" << node.second << endl;
  }
  {
    vector<string> gold{ "a", "aa", "aaaa", "abb" };
    {
      // Forward
      auto git = gold.begin();
      for (auto it = ots.begin(); it != ots.end(); ++it) {
        const auto key = ots.path_to(it);
        assert(string(begin(key), end(key)) == *git++);
      }
      assert(git == gold.end());
    }
    {
      // Reverse
      auto git = gold.rbegin();
      for (auto it = ots.rbegin(); it != ots.rend(); ++it) {
        const auto key = ots.path_to(it);
        assert(string(begin(key), end(key)) == *git++);
      }
      assert(git == gold.rend());
    }
  }
  
  return EXIT_SUCCESS;
}