void CTestCInputDataSet::getInputImagesIds()
{
CInputDataSet inputDataSet(pathToInputDataSet);
data = inputDataSet.getInputImages();
QCOMPARE(std::get<0>((*data)[0]), (uint32_t)0);
QCOMPARE(std::get<0>((*data)[1]), (uint32_t)1);
QCOMPARE(std::get<0>((*data)[2]), (uint32_t)2);
}
void CTestCInputDataSet::getInputImagesQImages()
{
CInputDataSet inputDataSet(pathToInputDataSet);
data = inputDataSet.getInputImages();
QCOMPARE(std::get<1>((*data)[0]), testImage0);
QCOMPARE(std::get<1>((*data)[1]), testImage1);
QCOMPARE(std::get<1>((*data)[2]), testImage2);
}
int main(int argc, char** argv) {
std::string trainingDir = "data//training";
std::string doc = "data//test//posting.txt";
/*
FileTokenizer tokenizer(doc);
while(tokenizer.hasNext()) {
std::string token = tokenizer.nextToken();
std::cout<<token<<std::endl;
}
*/
InputDataSet inputDataSet(trainingDir);
OfflineNaiveBayesClassifier offlineNBC(inputDataSet);
offlineNBC.classify(doc);
return 0;
}
void CTestCInputDataSet::getInputImagesCImagePreviewItems()
{
CInputDataSet inputDataSet(pathToInputDataSet);
data = inputDataSet.getInputImages();
CImagePreviewItem imagePrewviewItem0 = std::get<2>((*data)[0]);
CImagePreviewItem imagePrewviewItem1 = std::get<2>((*data)[1]);
CImagePreviewItem imagePrewviewItem2 = std::get<2>((*data)[2]);
//Compare the QIcons
QCOMPARE(imagePrewviewItem0.icon().pixmap(500,500).toImage(), testImage0);
QCOMPARE(imagePrewviewItem1.icon().pixmap(500,500).toImage(), testImage1);
QCOMPARE(imagePrewviewItem2.icon().pixmap(500,500).toImage(), testImage2);
//Compare the imagePrewviewItemIds
QCOMPARE(imagePrewviewItem0.getImageId(), (uint32_t)0);
QCOMPARE(imagePrewviewItem1.getImageId(), (uint32_t)1);
QCOMPARE(imagePrewviewItem2.getImageId(), (uint32_t)2);
//Compage the file namens
QCOMPARE(imagePrewviewItem0.text(), QString("0.png"));
QCOMPARE(imagePrewviewItem1.text(), QString("1.png"));
QCOMPARE(imagePrewviewItem2.text(), QString("2.png"));
}
