Unverified Commit 2e42fe27 authored by Scott Cyphers's avatar Scott Cyphers Committed by GitHub

Merge branch 'master' into cyphers/adiffinout

parents 125b1a85 5fe18494
...@@ -31,7 +31,8 @@ op::RequiresTensorViewArgs::RequiresTensorViewArgs(const std::string& node_type, ...@@ -31,7 +31,8 @@ op::RequiresTensorViewArgs::RequiresTensorViewArgs(const std::string& node_type,
{ {
if (nullptr == std::dynamic_pointer_cast<const TensorViewType>(arg->get_value_type())) if (nullptr == std::dynamic_pointer_cast<const TensorViewType>(arg->get_value_type()))
{ {
throw ngraph_error("Arguments must be tensor views"); throw ngraph_error("Arguments for node type \"" + node_type +
"\" must be tensor views");
} }
} }
} }
...@@ -75,7 +75,8 @@ TEST(type_prop, broadcast_bad_arguments) ...@@ -75,7 +75,8 @@ TEST(type_prop, broadcast_bad_arguments)
} }
catch (const ngraph_error& error) catch (const ngraph_error& error)
{ {
EXPECT_EQ(error.what(), std::string("Arguments must be tensor views")); EXPECT_EQ(error.what(),
std::string("Arguments for node type \"Broadcast\" must be tensor views"));
} }
catch (...) catch (...)
{ {
...@@ -382,7 +383,8 @@ TEST(type_prop, dot_deduce_reduction_axes_size_mismatch) ...@@ -382,7 +383,8 @@ TEST(type_prop, dot_deduce_reduction_axes_size_mismatch)
// //
// Tests for binary elementwise ops. // Tests for binary elementwise ops.
// //
void test_binary(shared_ptr<Node>(f)(const shared_ptr<Node>& x, const shared_ptr<Node>& y)) void test_binary(std::string node_type,
shared_ptr<Node>(f)(const shared_ptr<Node>& x, const shared_ptr<Node>& y))
{ {
// Check for bad arguments // Check for bad arguments
auto tp0_param = make_shared<op::Parameter>(make_shared<TupleType>()); auto tp0_param = make_shared<op::Parameter>(make_shared<TupleType>());
...@@ -407,7 +409,9 @@ void test_binary(shared_ptr<Node>(f)(const shared_ptr<Node>& x, const shared_ptr ...@@ -407,7 +409,9 @@ void test_binary(shared_ptr<Node>(f)(const shared_ptr<Node>& x, const shared_ptr
} }
catch (const ngraph_error& error) catch (const ngraph_error& error)
{ {
EXPECT_EQ(error.what(), std::string("Arguments must be tensor views")); EXPECT_EQ(
error.what(),
std::string("Arguments for node type \"" + node_type + "\" must be tensor views"));
} }
catch (...) catch (...)
{ {
...@@ -469,30 +473,34 @@ void test_binary(shared_ptr<Node>(f)(const shared_ptr<Node>& x, const shared_ptr ...@@ -469,30 +473,34 @@ void test_binary(shared_ptr<Node>(f)(const shared_ptr<Node>& x, const shared_ptr
TEST(type_prop, add_bad_arguments) TEST(type_prop, add_bad_arguments)
{ {
test_binary([](const shared_ptr<Node>& x, const shared_ptr<Node>& y) -> shared_ptr<Node> { test_binary("Add",
return make_shared<op::Add>(x, y); [](const shared_ptr<Node>& x, const shared_ptr<Node>& y) -> shared_ptr<Node> {
}); return make_shared<op::Add>(x, y);
});
} }
TEST(type_prop, divide_bad_arguments) TEST(type_prop, divide_bad_arguments)
{ {
test_binary([](const shared_ptr<Node>& x, const shared_ptr<Node>& y) -> shared_ptr<Node> { test_binary("Divide",
return make_shared<op::Divide>(x, y); [](const shared_ptr<Node>& x, const shared_ptr<Node>& y) -> shared_ptr<Node> {
}); return make_shared<op::Divide>(x, y);
});
} }
TEST(type_prop, multiply_bad_arguments) TEST(type_prop, multiply_bad_arguments)
{ {
test_binary([](const shared_ptr<Node>& x, const shared_ptr<Node>& y) -> shared_ptr<Node> { test_binary("Multiply",
return make_shared<op::Multiply>(x, y); [](const shared_ptr<Node>& x, const shared_ptr<Node>& y) -> shared_ptr<Node> {
}); return make_shared<op::Multiply>(x, y);
});
} }
TEST(type_prop, subtract_bad_arguments) TEST(type_prop, subtract_bad_arguments)
{ {
test_binary([](const shared_ptr<Node>& x, const shared_ptr<Node>& y) -> shared_ptr<Node> { test_binary("Subtract",
return make_shared<op::Subtract>(x, y); [](const shared_ptr<Node>& x, const shared_ptr<Node>& y) -> shared_ptr<Node> {
}); return make_shared<op::Subtract>(x, y);
});
} }
TEST(type_prop, comparison_good) TEST(type_prop, comparison_good)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment