experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[Grappler] Update templating in GraphView test.

Browse Source 
PiperOrigin-RevId: 249374323
This commit is contained in:
Andy Ly 2019-05-21 19:54:37 -07:00 committed by TensorFlower Gardener
parent 0ba1667fde
commit 74f30571be
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tensorflow/core/grappler/utils/graph_view_test.cc
View File

@ -48,13 +48,15 @@ GraphDef SimpleTestGraph() {
}
template <typename T>
class TypedGraphViewTest : public ::testing::Test {
public:
const string type_as_string_ =
std::is_same<T, GraphView>::value ? "GraphView" : "MutableGraphView";
};
const string GetGraphViewTypeAsString() {
return std::is_same<T, class GraphView>::value ? "GraphView"
: "MutableGraphView";
}
using GraphViewTypes = ::testing::Types<GraphView, MutableGraphView>;
template <typename T>
class TypedGraphViewTest : public ::testing::Test {};
TYPED_TEST_SUITE(TypedGraphViewTest, GraphViewTypes);
TYPED_TEST(TypedGraphViewTest, GraphWithDuplicateNodeNames) {
@ -67,7 +69,7 @@ TYPED_TEST(TypedGraphViewTest, GraphWithDuplicateNodeNames) {
EXPECT_EQ(s.error_message(),
absl::Substitute(
"$0::$0 error: graph has multiple nodes with the name 'a'.",
this->type_as_string_));
GetGraphViewTypeAsString<TypeParam>()));
}
TYPED_TEST(TypedGraphViewTest, GraphWithMissingFanins) {
@ -78,7 +80,7 @@ TYPED_TEST(TypedGraphViewTest, GraphWithMissingFanins) {
EXPECT_FALSE(s.ok());
EXPECT_EQ(s.error_message(),
absl::Substitute("$0::$0 error: node 'a' has missing fanin 'b:3'.",
this->type_as_string_));
GetGraphViewTypeAsString<TypeParam>()));
}
TYPED_TEST(TypedGraphViewTest, GraphWithSelfCycles) {
@ -90,7 +92,7 @@ TYPED_TEST(TypedGraphViewTest, GraphWithSelfCycles) {
EXPECT_EQ(
s.error_message(),
absl::Substitute("$0::$0 error: node 'a' has self cycle fanin 'a:4'.",
this->type_as_string_));
GetGraphViewTypeAsString<TypeParam>()));
}
TYPED_TEST(TypedGraphViewTest, GraphWithMisorderedFanins) {
@ -103,7 +105,7 @@ TYPED_TEST(TypedGraphViewTest, GraphWithMisorderedFanins) {
EXPECT_EQ(s.error_message(),
absl::Substitute("$0::$0 error: node 'a' has regular fanin 'b:4' "
"after controlling fanins.",
this->type_as_string_));
GetGraphViewTypeAsString<TypeParam>()));
}
TYPED_TEST(TypedGraphViewTest, GetNodeWithIndex) {