test_option.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_BINDINGS_TESTS_TEST_OPTION_HPP
#define MLPACK_CORE_BINDINGS_TESTS_TEST_OPTION_HPP

#include 

#include <mlpack/core/util/io.hpp>
#include "get_printable_param.hpp"
#include "get_param.hpp"
#include "get_allocated_memory.hpp"
#include "delete_allocated_memory.hpp"

namespace mlpack {
namespace bindings {
namespace tests {

// Defined in mlpack_main.hpp.
extern std::string programName;

template<typename N>
class TestOption
{
 public:
  TestOption(const N defaultValue,
             const std::string& identifier,
             const std::string& description,
             const std::string& alias,
             const std::string& cppName,
             const bool required = false,
             const bool input = true,
             const bool noTranspose = false,
             const std::string& testName = "")
  {
    // Create the ParamData object to give to IO.
    util::ParamData data;

    data.desc = description;
    data.name = identifier;
    data.tname = TYPENAME(N);
    data.alias = alias[0];
    data.wasPassed = false;
    data.noTranspose = noTranspose;
    data.required = required;
    data.input = input;
    data.loaded = false;
    data.cppType = cppName;
    data.value = boost::any(defaultValue);
    data.persistent = false;

    const std::string tname = data.tname;

    IO::RestoreSettings(testName, false);

    // Set some function pointers that we need.
    IO::GetSingleton().functionMap[tname]["GetPrintableParam"] =
        &GetPrintableParam;
    IO::GetSingleton().functionMap[tname]["GetParam"] = &GetParam;
    IO::GetSingleton().functionMap[tname]["GetAllocatedMemory"] =
        &GetAllocatedMemory;
    IO::GetSingleton().functionMap[tname]["DeleteAllocatedMemory"] =
        &DeleteAllocatedMemory;

    IO::Add(std::move(data));

    // If this is an output option, set it as passed.
    if (!input)
      IO::SetPassed(identifier);

    IO::StoreSettings(testName);
    IO::ClearSettings();
  }
};

} // namespace tests
} // namespace bindings
} // namespace mlpack

#endif