2012-02-08 21:47:54 +04:00
|
|
|
#include "Data.h"
|
|
|
|
|
#include "Scorer.h"
|
|
|
|
|
#include "ScorerFactory.h"
|
|
|
|
|
|
|
|
|
|
#define BOOST_TEST_MODULE MertData
|
|
|
|
|
#include <boost/test/unit_test.hpp>
|
|
|
|
|
|
2012-02-20 06:04:21 +04:00
|
|
|
#include <boost/scoped_ptr.hpp>
|
2012-02-08 21:47:54 +04:00
|
|
|
|
|
|
|
|
//very basic test of sharding
|
|
|
|
|
BOOST_AUTO_TEST_CASE(shard_basic) {
|
2012-02-20 06:04:21 +04:00
|
|
|
boost::scoped_ptr<Scorer> scorer(ScorerFactory::getScorer("BLEU", ""));
|
2012-02-08 21:47:54 +04:00
|
|
|
Data data(*scorer);
|
|
|
|
|
FeatureArray fa1, fa2, fa3, fa4;
|
|
|
|
|
ScoreArray sa1, sa2, sa3, sa4;
|
|
|
|
|
fa1.setIndex("1");
|
|
|
|
|
fa2.setIndex("2");
|
|
|
|
|
fa3.setIndex("3");
|
|
|
|
|
fa4.setIndex("4");
|
|
|
|
|
sa1.setIndex("1");
|
|
|
|
|
sa2.setIndex("2");
|
|
|
|
|
sa3.setIndex("3");
|
|
|
|
|
sa4.setIndex("4");
|
|
|
|
|
data.getFeatureData()->add(fa1);
|
|
|
|
|
data.getFeatureData()->add(fa2);
|
|
|
|
|
data.getFeatureData()->add(fa3);
|
|
|
|
|
data.getFeatureData()->add(fa4);
|
|
|
|
|
data.getScoreData()->add(sa1);
|
|
|
|
|
data.getScoreData()->add(sa2);
|
|
|
|
|
data.getScoreData()->add(sa3);
|
|
|
|
|
data.getScoreData()->add(sa4);
|
|
|
|
|
|
|
|
|
|
vector<Data> shards;
|
|
|
|
|
data.createShards(2,0,"",shards);
|
|
|
|
|
|
|
|
|
|
BOOST_CHECK_EQUAL(shards.size(),2);
|
|
|
|
|
BOOST_CHECK_EQUAL(shards[1].getFeatureData()->size(),2);
|
|
|
|
|
}
|
