From 26e2488b81bffe544f641eeca13e0f3a99964752 Mon Sep 17 00:00:00 2001
From: fishbell <bell.song@intel.com>
Date: Tue, 22 Jun 2021 11:34:09 -0400
Subject: [PATCH] use parametrized vistor API

---
 ngraph/test/visitors/op/atan.cpp | 27 ++++++---------------------
 1 file changed, 6 insertions(+), 21 deletions(-)

diff --git a/ngraph/test/visitors/op/atan.cpp b/ngraph/test/visitors/op/atan.cpp
index c31a7780838106..927211a283da73 100644
--- a/ngraph/test/visitors/op/atan.cpp
+++ b/ngraph/test/visitors/op/atan.cpp
@@ -2,26 +2,11 @@
 // SPDX-License-Identifier: Apache-2.0
 //
 
-#include "gtest/gtest.h"
+#include "unary_ops.hpp"
 
-#include "ngraph/ngraph.hpp"
-#include "ngraph/op/util/attr_types.hpp"
-#include "ngraph/opsets/opset1.hpp"
+using Types = ::testing::Types<UnaryOperatorType<ngraph::op::v0::Atan, element::f32>>;
 
-#include "util/visitor.hpp"
-
-using namespace std;
-using namespace ngraph;
-using ngraph::test::NodeBuilder;
-
-TEST(attributes, atan_op)
-{
-    NodeBuilder::get_ops().register_factory<op::Atan>();
-    const auto data_node = make_shared<op::Parameter>(element::f32, Shape{1});
-    const auto atan = make_shared<op::Atan>(data_node);
-
-    const NodeBuilder builder(atan);
-    const auto atan_attr_number = 0;
-
-    EXPECT_EQ(builder.get_value_map_size(), atan_attr_number);
-}
+INSTANTIATE_TYPED_TEST_CASE_P(visitor_without_attribute,
+                              UnaryOperatorVisitor,
+                              Types,
+                              UnaryOperatorTypeName);
\ No newline at end of file