dnn-layer-math-unary-test: add unit test for atan

Signed-off-by: Ting Fu <ting.fu@intel.com>
Signed-off-by: Guo Yejun <yejun.guo@intel.com>

diff --git a/tests/dnn/dnn-layer-mathunary-test.c b/tests/dnn/dnn-layer-mathunary-test.c
index 540ea4cef531e373b99ea762d666e672045effd2..bf77c44bbeb5234bcb1f65767c44e99bdf96c78a 100644 (file)
--- a/tests/dnn/dnn-layer-mathunary-test.c
+++ b/tests/dnn/dnn-layer-mathunary-test.c
@@ -42,6 +42,8 @@ static float get_expected(float f, DNNMathUnaryOperation op)
         return asin(f);
     case DMUO_ACOS:
         return acos(f);
+    case DMUO_ATAN:
+        return atan(f);
     default:
         av_assert0(!"not supported yet");
         return 0.f;
@@ -97,5 +99,7 @@ int main(int agrc, char **argv)
         return 1;
     if (test(DMUO_ACOS))
         return 1;
+    if (test(DMUO_ATAN))
+        return 1;
     return 0;
 }