Commit 4446ca3a authored by Khalique's avatar Khalique
Browse files

fix test naming and order

parent 8c1f0289
...@@ -58,7 +58,7 @@ TEST_CASE(add_scalar_test) ...@@ -58,7 +58,7 @@ TEST_CASE(add_scalar_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(argmax) TEST_CASE(argmax_test)
{ {
migraphx::program p; migraphx::program p;
auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}}); auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
...@@ -69,7 +69,7 @@ TEST_CASE(argmax) ...@@ -69,7 +69,7 @@ TEST_CASE(argmax)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(argmin) TEST_CASE(argmin_test)
{ {
migraphx::program p; migraphx::program p;
auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}}); auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
...@@ -841,6 +841,16 @@ TEST_CASE(reshape_non_standard_test) ...@@ -841,6 +841,16 @@ TEST_CASE(reshape_non_standard_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(round_test)
{
migraphx::program p;
auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::double_type, {10, 5}});
p.add_instruction(migraphx::op::round{}, input);
auto prog = migraphx::parse_onnx("round_test.onnx");
EXPECT(p == prog);
}
TEST_CASE(shape_test) TEST_CASE(shape_test)
{ {
migraphx::program p; migraphx::program p;
...@@ -1054,14 +1064,4 @@ TEST_CASE(unknown_test) ...@@ -1054,14 +1064,4 @@ TEST_CASE(unknown_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(round_test)
{
migraphx::program p;
auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::double_type, {10, 5}});
p.add_instruction(migraphx::op::round{}, input);
auto prog = migraphx::parse_onnx("round_test.onnx");
EXPECT(p == prog);
}
int main(int argc, const char* argv[]) { test::run(argc, argv); } int main(int argc, const char* argv[]) { test::run(argc, argv); }
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