MasterTrainerTest Class Reference

Inheritance diagram for MasterTrainerTest:

Protected Member Functions
void	SetUp () override
std::string	TestDataNameToPath (const std::string &name)
std::string	TmpNameToPath (const std::string &name)
	MasterTrainerTest ()
	~MasterTrainerTest () override
void	LoadMasterTrainer ()
void	VerifyIl1 ()
Protected Member Functions inherited from testing::Test
	Test ()
virtual void	SetUp ()
virtual void	TearDown ()

Protected Attributes
ShapeTable *	shape_table_
std::unique_ptr< MasterTrainer >	master_trainer_

Additional Inherited Members
Public Member Functions inherited from testing::Test
virtual	~Test ()
Static Public Member Functions inherited from testing::Test
static void	SetUpTestSuite ()
static void	TearDownTestSuite ()
static void	TearDownTestCase ()
static void	SetUpTestCase ()
static bool	HasFatalFailure ()
static bool	HasNonfatalFailure ()
static bool	IsSkipped ()
static bool	HasFailure ()
static void	RecordProperty (const std::string &key, const std::string &value)
static void	RecordProperty (const std::string &key, int value)

Detailed Description

Definition at line 144 of file mastertrainer_test.cc.

Constructor & Destructor Documentation

MasterTrainerTest()

MasterTrainerTest::MasterTrainerTest ( )

inlineprotected

Definition at line 159 of file mastertrainer_test.cc.

                      {
    shape_table_ = nullptr;
    master_trainer_ = nullptr;
  }

~MasterTrainerTest()

MasterTrainerTest::~MasterTrainerTest ( )

inlineoverrideprotected

Definition at line 163 of file mastertrainer_test.cc.

                                {
    delete shape_table_;
  }

Member Function Documentation

LoadMasterTrainer()

void MasterTrainerTest::LoadMasterTrainer ( )

inlineprotected

Definition at line 170 of file mastertrainer_test.cc.

                           {
    FLAGS_output_trainer = TmpNameToPath("tmp_trainer").c_str();
    FLAGS_F = file::JoinPath(LANGDATA_DIR, "font_properties").c_str();
    FLAGS_X = TestDataNameToPath("eng.xheights").c_str();
    FLAGS_U = TestDataNameToPath("eng.unicharset").c_str();
    std::string tr_file_name(TestDataNameToPath("eng.Arial.exp0.tr"));
    const char *filelist[] = {tr_file_name.c_str(), nullptr};
    std::string file_prefix;
    delete shape_table_;
    shape_table_ = nullptr;
    master_trainer_ = LoadTrainingData(filelist, false, &shape_table_, file_prefix);
    EXPECT_TRUE(master_trainer_ != nullptr);
    EXPECT_TRUE(shape_table_ != nullptr);
  }

SetUp()

void MasterTrainerTest::SetUp ( )

inlineoverrideprotectedvirtual

Reimplemented from testing::Test.

Definition at line 147 of file mastertrainer_test.cc.

                        {
    std::locale::global(std::locale(""));
    file::MakeTmpdir();
  }

TestDataNameToPath()

std::string MasterTrainerTest::TestDataNameToPath ( const std::string &  name )

inlineprotected

Definition at line 152 of file mastertrainer_test.cc.

                                                      {
    return file::JoinPath(TESTING_DIR, name);
  }

TmpNameToPath()

std::string MasterTrainerTest::TmpNameToPath ( const std::string &  name )

inlineprotected

Definition at line 155 of file mastertrainer_test.cc.

                                                 {
    return file::JoinPath(FLAGS_test_tmpdir, name);
  }

VerifyIl1()

void MasterTrainerTest::VerifyIl1 ( )

inlineprotected

Definition at line 187 of file mastertrainer_test.cc.

                   {
    // Find the font id for Arial.
    int font_id = master_trainer_->GetFontInfoId("Arial");
    EXPECT_GE(font_id, 0);
    // Track down the characters we are interested in.
    int unichar_I = master_trainer_->unicharset().unichar_to_id("I");
    EXPECT_GT(unichar_I, 0);
    int unichar_l = master_trainer_->unicharset().unichar_to_id("l");
    EXPECT_GT(unichar_l, 0);
    int unichar_1 = master_trainer_->unicharset().unichar_to_id("1");
    EXPECT_GT(unichar_1, 0);
    // Now get the shape ids.
    int shape_I = shape_table_->FindShape(unichar_I, font_id);
    EXPECT_GE(shape_I, 0);
    int shape_l = shape_table_->FindShape(unichar_l, font_id);
    EXPECT_GE(shape_l, 0);
    int shape_1 = shape_table_->FindShape(unichar_1, font_id);
    EXPECT_GE(shape_1, 0);
 
    float dist_I_l = master_trainer_->ShapeDistance(*shape_table_, shape_I, shape_l);
    // No tolerance here. We expect that I and l should match exactly.
    EXPECT_EQ(0.0f, dist_I_l);
    float dist_l_I = master_trainer_->ShapeDistance(*shape_table_, shape_l, shape_I);
    // BOTH ways.
    EXPECT_EQ(0.0f, dist_l_I);
 
    // l/1 on the other hand should be distinct.
    float dist_l_1 = master_trainer_->ShapeDistance(*shape_table_, shape_l, shape_1);
    EXPECT_GT(dist_l_1, kMin1lDistance);
    float dist_1_l = master_trainer_->ShapeDistance(*shape_table_, shape_1, shape_l);
    EXPECT_GT(dist_1_l, kMin1lDistance);
 
    // So should I/1.
    float dist_I_1 = master_trainer_->ShapeDistance(*shape_table_, shape_I, shape_1);
    EXPECT_GT(dist_I_1, kMin1lDistance);
    float dist_1_I = master_trainer_->ShapeDistance(*shape_table_, shape_1, shape_I);
    EXPECT_GT(dist_1_I, kMin1lDistance);
  }

Member Data Documentation

master_trainer_

std::unique_ptr<MasterTrainer> MasterTrainerTest::master_trainer_

protected

Definition at line 228 of file mastertrainer_test.cc.

shape_table_

ShapeTable* MasterTrainerTest::shape_table_

protected

Definition at line 227 of file mastertrainer_test.cc.

---

The documentation for this class was generated from the following file:

• /media/home/debian/src/github/tesseract-ocr/tesseract/unittest/mastertrainer_test.cc