test_function_map.hpp

Go to the documentation of this file.

#ifndef MLPACK_BINDINGS_TESTS_TEST_FUNCTION_MAP_HPP
#define MLPACK_BINDINGS_TESTS_TEST_FUNCTION_MAP_HPP

#include <mlpack/core/util/param_data.hpp>

namespace mlpack {
namespace bindings {
namespace tests {

class TestFunctionMap
{
 public:
  // Convenience typedef.
  typedef std::map<std::string, std::map<std::string,
      void (*)(util::ParamData&, const void*, void*)>> FunctionMapType;

  static TestFunctionMap& GetSingleton();

  static void RegisterFunction(
      const std::string& tname,
      const std::string& functionName,
      void (*func)(util::ParamData&, const void*, void*));

  static const FunctionMapType& FunctionMap();

 private:
  TestFunctionMap();

  FunctionMapType functionMap;
};

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

#endif