Added tests

ivy-llc · Sep 14, 2023 · 83b04a0 · 83b04a0
1 parent 555a298
commit 83b04a0
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 1 deletion.
diff --git a/ivy/functional/ivy/experimental/layers.py b/ivy/functional/ivy/experimental/layers.py
@@ -431,7 +431,7 @@ def avg_pool2d(
              [ 3.,  4.],
              [ 4.,  5.]]],
 
-
+             
            [[[ 7.,  8.],
              [ 9., 10.],
              [10., 11.]]]])

diff --git a/ivy_tests/test_ivy/test_frontends/test_paddle/test_fft.py b/ivy_tests/test_ivy/test_frontends/test_paddle/test_fft.py
@@ -107,6 +107,40 @@ def test_paddle_hfft(
     )
 
 
+@handle_frontend_test(
+    fn_tree="paddle.fft.hfftn",
+    dtype_x_axis=helpers.dtype_values_axis(
+        available_dtypes=helpers.get_dtypes("complex"),
+        min_value=-10,
+        max_value=10,
+        min_num_dims=1,
+        valid_axis=True,
+        force_int_axis=True,
+    ),
+)
+def test_paddle_hfftn(
+    dtype_x_axis,
+    frontend,
+    test_flags,
+    fn_tree,
+    on_device,
+    backend_fw,
+):
+    input_dtype, x, s, axes = dtype_x_axis
+    helpers.test_frontend_function(
+        input_dtypes=input_dtype,
+        backend_to_test=backend_fw,
+        frontend=frontend,
+        test_flags=test_flags,
+        fn_tree=fn_tree,
+        on_device=on_device,
+        test_values=True,
+        x=x,
+        s=s,
+        axes=axes,
+    )
+
+
 @handle_frontend_test(
     fn_tree="paddle.fft.ifft",
     dtype_x_axis=helpers.dtype_values_axis(