Not using BaseTest legacy code anymore

Put all the testing code in the Test()

Author: Nerdyvedi

modules/ximgproc/test/test_globalmatting.cpp

@@ -2,7 +2,6 @@
 // It is subject to the license terms in the LICENSE file found in the top-level directory
 // of this distribution and at http://opencv.org/license.html.
 #include "test_precomp.hpp"
-using namespace cv;
 
 namespace opencv_test
 {
@@ -13,71 +12,36 @@ namespace opencv_test
     const std::string TRIMAP_FILENAME = "trimap/doll.png";
 
 
-    class CV_GlobalMattingTest : public cvtest::BaseTest
+    void runModel()
     {
-      public:
-         CV_GlobalMattingTest();
-
-      protected:
-         Ptr<GlobalMatting> gm;
-         virtual void run(int);
-         void runModel();
-
-    };
-
-    void CV_GlobalMattingTest::runModel()
-    {
-      std::string folder       = std::string(cvtest::TS::ptr()->get_data_path());
-      std::string img_path     = folder + INPUT_DIR + "/" + IMAGE_FILENAME;
-      std::string trimap_path  = folder + INPUT_DIR + "/" + TRIMAP_FILENAME;
+      
+      Ptr<GlobalMatting> gm    = makePtr<GlobalMatting>();
+      std::string img_path     = cvtest::findDataFile(INPUT_DIR + "/" + IMAGE_FILENAME);
+      std::string trimap_path  = cvtest::findDataFile(INPUT_DIR + "/" + TRIMAP_FILENAME);
 
       Mat img     = cv::imread(img_path,cv::IMREAD_COLOR);
       Mat trimap  = cv::imread(trimap_path,cv::IMREAD_GRAYSCALE);
-      if(img.empty() || trimap.empty())
-      {
-        ts->printf(cvtest::TS::LOG,"Test images not found!\n");
-        ts->set_failed_test_info(cvtest::TS::FAIL_INVALID_TEST_DATA);
-        return;
-      }
-      if(img.cols!=trimap.cols || img.rows!=trimap.rows)
-      {
-        ts->printf(cvtest::TS::LOG,"Dimensions of trimap and the image are not the same");
-        ts->set_failed_test_info(cvtest::TS::FAIL_INVALID_TEST_DATA);
-        return;
-      }
+      ASSERT_FALSE(img.empty()) << "The Image could not be loaded: "<< img_path;
+      ASSERT_FALSE(trimap.empty()) << "The trimap could not be loaded: "<< trimap_path;
+      
+      ASSERT_EQ(img.cols,trimap.cols);
+      ASSERT_EQ(img.rows,trimap.rows);
       Mat foreground,alpha;
       int niter = 9;
-      this->gm->getMat(img,trimap,foreground,alpha,niter);
-      if(alpha.empty())
-      {
-        ts->printf(cvtest::TS::LOG,"Could not find the alpha matte for the image\n");
-        ts->set_failed_test_info(cvtest::TS::FAIL_BAD_ACCURACY);
-        return;
-      }
-
-      if(alpha.cols!=img.cols || alpha.rows!=img.rows)
-      {
-        ts->printf(cvtest::TS::LOG,"The dimensions of the output are not correct");
-        ts->set_failed_test_info(cvtest::TS::FAIL_BAD_ACCURACY);
-        return;
-      }
+      gm->getMat(img,trimap,foreground,alpha,niter);
+      
+      ASSERT_FALSE(foreground.empty()) << " Could not extract the foreground "  <<
+      ASSERT_FALSE(alpha.empty()) << " Could not generate alpha matte " <<  
+   
+      ASSERT_EQ(alpha.cols,img.cols)
+      ASSERT_EQ(alpha.rows,img.rows)  
+      
     }
 
-    CV_GlobalMattingTest::CV_GlobalMattingTest()
-    {
-      gm = makePtr<GlobalMatting>();
-    }
-    void CV_GlobalMattingTest::run(int)
-    {
-      runModel();
-    }
-
-
 
     TEST(CV_GlobalMattingTest,accuracy)
     {
-      CV_GlobalMattingTest test;
-      test.safe_run();
+      runModel();     
     }
 
   }