예제 #1
0
/**
 * @brief ANNOTATION search testcase.
 */
void ColorSearchTest::annotationSearchTest()
{
    SearchQuery query;
    query.setDatasets({"../../../test/testdata/SingleFrameVideo"});
    SearchObject obj;
    obj.setSourceDataset("../../../test/testdata/SingleFrameVideo");
    obj.setMedium(QDir("../../../test/testdata/SingleFrameVideo/SingleFrameVideo/").absolutePath());
    obj.setMediumIndex(1);
    RectangleAnnotation* ann = new RectangleAnnotation("11", "Person");
    ann->setX(315);
    ann->setY(53);
    ann->setWidth(169);
    ann->setHeight(570);
    obj.setAnnotation(ann);
    obj.setType(SearchObject::ANNOTATION);
    query.setSearchObject(obj);

    ColorSearch search;

    QList<DataPacket*> in;
    in.append(&query);
    bool ok = search.setInputs(in);
    QVERIFY(ok);

    QList<DataPacket*> out = search.run();
    QVERIFY(out.size() == 1);
}
예제 #2
0
/**
 * @brief MEDIUM search testcase.
 */
void ColorSearchTest::mediumSearchTest()
{
    SearchQuery query;
    query.setDatasets({"../../../test/testdata/Fotos"});
    SearchObject obj;
    obj.setSourceDataset("../../../test/testdata/Fotos");
    obj.setMedium(QDir("../../../test/testdata/Fotos/000_45.bmp").absolutePath());
    obj.setMediumIndex(0);
    obj.setType(SearchObject::MEDIUM);
    query.setSearchObject(obj);

    ColorSearch search;

    QList<DataPacket*> in;
    in.append(&query);
    bool ok = search.setInputs(in);
    QVERIFY(ok);

    QList<DataPacket*> out = search.run();
    QVERIFY(out.size() == 1);
}