add float type to accu test

Author: kuloPo

modules/ximgproc/test/test_radon_transform.cpp

@@ -13,20 +13,35 @@ TEST(RadonTransformTest, output_size)
     Mat radon;
     cv::ximgproc::RadonTransform(src, radon);
 
-    ASSERT_EQ(radon.rows, 180);
-    ASSERT_EQ(radon.cols, 256);
+    ASSERT_EQ(radon.rows, 363);
+    ASSERT_EQ(radon.cols, 180);
+
+    cv::ximgproc::RadonTransform(src, radon, 1, 0, 180, true);
+
+    ASSERT_EQ(radon.rows, 256);
+    ASSERT_EQ(radon.cols, 180);
 }
 
 TEST(RadonTransformTest, output_type)
 {
-    Mat src(Size(256, 256), CV_8U, Scalar(0));
-    circle(src, Point(128, 128), 64, Scalar(255), FILLED);
+    Mat src_int(Size(256, 256), CV_8U, Scalar(0));
+    circle(src_int, Point(128, 128), 64, Scalar(255), FILLED);
     Mat radon, radon_norm;
-    cv::ximgproc::RadonTransform(src, radon);
-    cv::ximgproc::RadonTransform(src, radon_norm, 1, 0, 180, false, true);
+    cv::ximgproc::RadonTransform(src_int, radon);
+    cv::ximgproc::RadonTransform(src_int, radon_norm, 1, 0, 180, false, true);
 
     ASSERT_EQ(radon.type(), CV_32SC1);
     ASSERT_EQ(radon_norm.type(), CV_8U);
+
+    Mat src_float(Size(256, 256), CV_32FC1, Scalar(0));
+    Mat src_double(Size(256, 256), CV_32FC1, Scalar(0));
+    cv::ximgproc::RadonTransform(src_float, radon);
+    cv::ximgproc::RadonTransform(src_float, radon_norm, 1, 0, 180, false, true);
+    ASSERT_EQ(radon.type(), CV_64FC1);
+    ASSERT_EQ(radon_norm.type(), CV_8U);
+    cv::ximgproc::RadonTransform(src_double, radon);
+    ASSERT_EQ(radon.type(), CV_64FC1);
+    ASSERT_EQ(radon_norm.type(), CV_8U);
 }
 
 TEST(RadonTransformTest, accuracy_by_pixel)
@@ -38,38 +53,27 @@ TEST(RadonTransformTest, accuracy_by_pixel)
 
     ASSERT_EQ(radon.at<int>(0, 0), 0);
 
-    ASSERT_GT(radon.at<int>(128, 128), 32000);
-    ASSERT_LT(radon.at<int>(128, 128), 33000);
+    ASSERT_GT(radon.at<int>(128, 128), 18000);
+    ASSERT_LT(radon.at<int>(128, 128), 19000);
 }
 
-TEST(RadonTransformTest, accuracy_by_col_sum)
+TEST(RadonTransformTest, accuracy_uchar)
 {
-    Mat src(Size(256, 256), CV_8U, Scalar(0));
-    circle(src, Point(128, 128), 64, Scalar(255), FILLED);
-    Mat radon;
-    cv::ximgproc::RadonTransform(src, radon, 1, 0, 180, false, true);
-    Mat sum_col;
-    cv::reduce(radon, sum_col, 0, REDUCE_SUM, CV_32SC1);
-
-    ASSERT_EQ(sum_col.at<int>(0), 0);
+    Mat src(Size(10, 10), CV_8UC1, Scalar(1));
+    cv::Mat radon;
+    ximgproc::RadonTransform(src, radon, 45, 0, 180, false, false);
 
-    ASSERT_GT(sum_col.at<int>(128), 45000);
-    ASSERT_LT(sum_col.at<int>(128), 46000);
+    ASSERT_EQ(sum(radon.col(0))[0], 100);
 }
 
-TEST(RadonTransformTest, accuracy_by_row_sum)
+TEST(RadonTransformTest, accuracy_float)
 {
-    Mat src(Size(256, 256), CV_8U, Scalar(0));
-    circle(src, Point(128, 128), 64, Scalar(255), FILLED);
-    Mat radon;
-    cv::ximgproc::RadonTransform(src, radon, 1, 0, 180, false, true);
-    Mat sum_row;
-    cv::reduce(radon, sum_row, 1, REDUCE_SUM, CV_32SC1);
-
-    ASSERT_LT(abs(sum_row.at<int>(0) - sum_row.at<int>(128)), 1000);
+    Mat src(Size(10, 10), CV_32FC1, Scalar(1.1));
+    cv::Mat radon;
+    ximgproc::RadonTransform(src, radon, 45, 0, 180, false, false);
 
-    ASSERT_GT(sum_row.at<int>(0), 25000);
-    ASSERT_LT(sum_row.at<int>(0), 26000);
+    ASSERT_GT(sum(radon.col(0))[0], 109);
+    ASSERT_LT(sum(radon.col(0))[0], 111);
 }
 
 } }