This repository has been archived by the owner on Apr 5, 2023. It is now read-only.
forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
init_test.cc
69 lines (56 loc) · 1.85 KB
/
init_test.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
#include <iostream>
#include <memory>
#include <gtest/gtest.h>
#include "caffe2/core/init.h"
#include "caffe2/core/logging.h"
namespace caffe2 {
namespace {
bool gTestInitFunctionHasBeenRun = false;
bool gTestFailInitFunctionHasBeenRun = false;
bool TestInitFunction(int*, char***) {
gTestInitFunctionHasBeenRun = true;
return true;
}
bool TestFailInitFunction(int*, char***) {
gTestFailInitFunctionHasBeenRun = true;
return false;
}
REGISTER_CAFFE2_INIT_FUNCTION(
TestInitFunction,
&TestInitFunction,
"Just a test to see if GlobalInit invokes "
"registered functions correctly.");
int dummy_argc = 1;
const char* dummy_name = "foo";
char** dummy_argv = const_cast<char**>(&dummy_name);
} // namespace
TEST(InitTest, TestInitFunctionHasRun) {
caffe2::GlobalInit(&dummy_argc, &dummy_argv);
EXPECT_TRUE(gTestInitFunctionHasBeenRun);
EXPECT_FALSE(gTestFailInitFunctionHasBeenRun);
}
TEST(InitTest, CanRerunGlobalInit) {
caffe2::GlobalInit(&dummy_argc, &dummy_argv);
EXPECT_TRUE(caffe2::GlobalInit(&dummy_argc, &dummy_argv));
}
void LateRegisterInitFunction() {
::caffe2::InitRegisterer testInitFunc(
TestInitFunction, false, "This should succeed but warn");
}
void LateRegisterEarlyInitFunction() {
::caffe2::InitRegisterer testSecondInitFunc(
TestInitFunction, true, "This should fail for early init");
}
void LateRegisterFailInitFunction() {
::caffe2::InitRegisterer testSecondInitFunc(
TestFailInitFunction, false, "This should fail for failed init");
}
TEST(InitTest, FailLateRegisterInitFunction) {
caffe2::GlobalInit(&dummy_argc, &dummy_argv);
LateRegisterInitFunction();
EXPECT_THROW(LateRegisterEarlyInitFunction(), ::c10::Error);
EXPECT_THROW(LateRegisterFailInitFunction(), ::c10::Error);
EXPECT_TRUE(gTestInitFunctionHasBeenRun);
EXPECT_TRUE(gTestFailInitFunctionHasBeenRun);
}
} // namespace caffe2