diff --git a/modules/gpu/test/test_main.cpp b/modules/gpu/test/test_main.cpp index 3420eb1ec0..89c1b272bc 100644 --- a/modules/gpu/test/test_main.cpp +++ b/modules/gpu/test/test_main.cpp @@ -1,6 +1,13 @@ #include "test_precomp.hpp" -CV_TEST_MAIN("gpu") - -// Run test with --gtest_catch_exceptions flag to avoid runtime errors in -// the case when there is no GPU +int main(int argc, char **argv) +{ + cvtest::TS::ptr()->init("gpu"); + ::testing::InitGoogleTest(&argc, argv); +#ifdef HAVE_CUDA + return RUN_ALL_TESTS(); +#else + std::cerr << "opencv_test_gpu: OpenCV was compiled without GPU support\n"; + return -1; +#endif +} \ No newline at end of file diff --git a/modules/gpu/test/test_meanshift.cpp b/modules/gpu/test/test_meanshift.cpp index e58bd15549..b297b982ac 100644 --- a/modules/gpu/test/test_meanshift.cpp +++ b/modules/gpu/test/test_meanshift.cpp @@ -125,11 +125,7 @@ struct CV_GpuMeanShiftTest : public cvtest::BaseTest }; -///////////////////////////////////////////////////////////////////////////// -/////////////////// tests registration ///////////////////////////////////// -///////////////////////////////////////////////////////////////////////////// - -CV_GpuMeanShiftTest CV_GpuMeanShift_test; +TEST(meanShift, accuracy) { CV_GpuMeanShiftTest test; test.safe_run(); } struct CV_GpuMeanShiftProcTest : public cvtest::BaseTest { diff --git a/modules/gpu/test/test_precomp.hpp b/modules/gpu/test/test_precomp.hpp index 9b048fc81e..89426c3bf0 100644 --- a/modules/gpu/test/test_precomp.hpp +++ b/modules/gpu/test/test_precomp.hpp @@ -2,6 +2,7 @@ #define __OPENCV_TEST_PRECOMP_HPP__ #include +#include #include "opencv2/core/core.hpp" #include "opencv2/highgui/highgui.hpp" #include "opencv2/calib3d/calib3d.hpp"