mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-28 20:11:22 +00:00
09806625cf
* Rename OrtInitialize to OrtCreateEnv in preparation for future. Add version number to structures * Forgot about exports * Update documentation
38 lines
1 KiB
C++
38 lines
1 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "core/session/onnxruntime_cxx_api.h"
|
|
#include <gtest/gtest.h>
|
|
|
|
#ifdef _WIN32
|
|
typedef const wchar_t* PATH_TYPE;
|
|
#define TSTR(X) L##X
|
|
#else
|
|
#define TSTR(X) (X)
|
|
typedef const char* PATH_TYPE;
|
|
#endif
|
|
|
|
//empty
|
|
static inline void ORT_API_CALL MyLoggingFunction(void*, OrtLoggingLevel, const char*, const char*, const char*, const char*) {
|
|
}
|
|
template <bool use_customer_logger>
|
|
class CApiTestImpl : public ::testing::Test {
|
|
protected:
|
|
OrtEnv* env = nullptr;
|
|
|
|
void SetUp() override {
|
|
if (use_customer_logger) {
|
|
ORT_THROW_ON_ERROR(OrtCreateEnvWithCustomLogger(MyLoggingFunction, nullptr, ORT_LOGGING_LEVEL_INFO, "Default", &env));
|
|
} else {
|
|
ORT_THROW_ON_ERROR(OrtCreateEnv(ORT_LOGGING_LEVEL_INFO, "Default", &env));
|
|
}
|
|
}
|
|
|
|
void TearDown() override {
|
|
if (env) OrtReleaseEnv(env);
|
|
}
|
|
|
|
// Objects declared here can be used by all tests in the test case for Foo.
|
|
};
|
|
|
|
typedef CApiTestImpl<false> CApiTest;
|