[FFmpeg-devel] [PATCH 6/6] dnn-layer-math-unary-test: add unit test for atan

[Ting Fu]

Signed-off-by: Ting Fu <ting.fu at intel.com>
---
 tests/dnn/dnn-layer-mathunary-test.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/dnn/dnn-layer-mathunary-test.c b/tests/dnn/dnn-layer-mathunary-test.c
index 540ea4cef5..bf77c44bbe 100644
--- 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;
 }
-- 
2.17.1