Add tests for bigger length

openvinotoolkit · Jan 24, 2025 · ee66cde · ee66cde
1 parent 13faa4d
commit ee66cde
Showing 1 changed file with 44 additions and 0 deletions.
diff --git a/src/plugins/template/tests/functional/op_reference/istft.cpp b/src/plugins/template/tests/functional/op_reference/istft.cpp
@@ -176,6 +176,28 @@ std::vector<ISTFTParams> generateISTFTParams() {
                         0.9590,  0.7095,  -0.2771, -0.9758, -0.6608, 0.3406,  0.9882,  0.6092,  -0.4027, -0.9962,
                         -0.5549, 0.4629,  0.9998,  0.4981,  -0.5211, -0.9989, -0.4390, 0.5769,  0.9936});
 
+    reference_tests::Tensor signal_55(
+        Shape{55},
+        ET,
+        std::vector<VT>{-0.9511, -0.1861, 0.7722,  0.9283,  0.1200,  -0.8129, -0.9014, -0.0534, 0.8500,  0.8704,
+                        -0.0134, -0.8833, -0.8356, 0.0801,  0.9126,  0.7971,  -0.1465, -0.9379, -0.7550, 0.2123,
+                        0.9590,  0.7095,  -0.2771, -0.9758, -0.6608, 0.3406,  0.9882,  0.6092,  -0.4027, -0.9962,
+                        -0.5549, 0.4629,  0.9998,  0.4981,  -0.5211, -0.9989, -0.4390, 0.5769,  0.9936,  0.3780,
+                        -0.6302, -0.9838, -0.3154, 0.6806,  0.9696,  0.2513,  -0.7281, -0.9511, -0.7281, 0.2513,
+                        0.9696,  0.6806,  -0.3154, -0.9838, -0.6302});
+
+    reference_tests::Tensor signal_60(
+        Shape{60},
+        ET,
+        std::vector<VT>{-0.9511, -0.1861,  0.7722,  0.9283,  0.1200, -0.8129, -0.9014, -0.0534,
+         0.8500,  0.8704, -0.0134, -0.8833, -0.8356,  0.0801,  0.9126,  0.7971,
+        -0.1465, -0.9379, -0.7550,  0.2123,  0.9590,  0.7095, -0.2771, -0.9758,
+        -0.6608,  0.3406,  0.9882,  0.6092, -0.4027, -0.9962, -0.5549,  0.4629,
+         0.9998,  0.4981, -0.5211, -0.9989, -0.4390,  0.5769,  0.9936,  0.3780,
+        -0.6302, -0.9838, -0.3154,  0.6806,  0.9696,  0.2513, -0.7281, -0.9511,
+        -0.7281,  0.2513,  0.9696,  0.6806, -0.3154, -0.9838, -0.6302,  0.3780,
+         0.0000,  0.0000,  0.0000,  0.0000});
+
     reference_tests::Tensor signal_1_48(
         signal_1_48_shape,
         ET,
@@ -364,6 +386,8 @@ std::vector<ISTFTParams> generateISTFTParams() {
     reference_tests::Tensor length_16(Shape{}, IT, std::vector<INT_T>{16});
     reference_tests::Tensor length_39(Shape{}, IT, std::vector<INT_T>{39});
     reference_tests::Tensor length_48(Shape{}, IT, std::vector<INT_T>{48});
+    reference_tests::Tensor length_55(Shape{}, IT, std::vector<INT_T>{55});
+    reference_tests::Tensor length_60(Shape{}, IT, std::vector<INT_T>{60});
 
     std::vector<ISTFTParams> params;
     params.emplace_back(output_stft_9_1_2_transp_win_one,
@@ -466,6 +490,26 @@ std::vector<ISTFTParams> generateISTFTParams() {
                         signal_39,
                         "basic_1D_transp_two_win_step_16_center_norm_length_39");
 
+    params.emplace_back(output_stft_9_4_2_transp_win_two_center_norm,
+                        two_window_16,
+                        frame_size_16,
+                        frame_step_16,
+                        length_55,
+                        true,
+                        true,
+                        signal_55,
+                        "basic_1D_transp_two_win_step_16_center_norm_length_55");
+
+    params.emplace_back(output_stft_9_4_2_transp_win_two_center_norm,
+                        two_window_16,
+                        frame_size_16,
+                        frame_step_16,
+                        length_60,
+                        true,
+                        true,
+                        signal_60,
+                        "basic_1D_transp_two_win_step_16_center_norm_length_60");
+
     return params;
 }