From 24d1781cbd72468146f5361687ab6925a5f791fe Mon Sep 17 00:00:00 2001 From: Ting Fu Date: Thu, 18 Jun 2020 17:15:36 +0800 Subject: [PATCH] dnn-layer-math-unary-test: add unit test for atan Signed-off-by: Ting Fu Signed-off-by: Guo Yejun --- 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; }