diff options
author | Hieu Hoang <hieuhoang@gmail.com> | 2012-11-12 23:56:18 +0400 |
---|---|---|
committer | Hieu Hoang <hieuhoang@gmail.com> | 2012-11-12 23:56:18 +0400 |
commit | 5e3ef23cef6101d2c098eb3445f562e8f595655b (patch) | |
tree | b8c332b6fa82bae84ea4910967a10ba1b08a7107 /moses/ScoreComponentCollectionTest.cpp | |
parent | 8c785cff2b1be3cccd76ea9026f71b649762dfc3 (diff) |
move moses/src/* to moses/
Diffstat (limited to 'moses/ScoreComponentCollectionTest.cpp')
-rw-r--r-- | moses/ScoreComponentCollectionTest.cpp | 149 |
1 files changed, 149 insertions, 0 deletions
diff --git a/moses/ScoreComponentCollectionTest.cpp b/moses/ScoreComponentCollectionTest.cpp new file mode 100644 index 000000000..905358269 --- /dev/null +++ b/moses/ScoreComponentCollectionTest.cpp @@ -0,0 +1,149 @@ +/*********************************************************************** +Moses - factored phrase-based language decoder +Copyright (C) 2010 University of Edinburgh + +This library is free software; you can redistribute it and/or +modify it under the terms of the GNU Lesser General Public +License as published by the Free Software Foundation; either +version 2.1 of the License, or (at your option) any later version. + +This library is distributed in the hope that it will be useful, +but WITHOUT ANY WARRANTY; without even the implied warranty of +MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +Lesser General Public License for more details. + +You should have received a copy of the GNU Lesser General Public +License along with this library; if not, write to the Free Software +Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA +***********************************************************************/ + +#include <stdexcept> + +#include <boost/test/unit_test.hpp> + +#include "DummyScoreProducers.h" +#include "FeatureFunction.h" +#include "ScoreComponentCollection.h" + +using namespace Moses; +using namespace std; + +BOOST_AUTO_TEST_SUITE(scc) + +class MockStatelessFeatureFunction : public StatelessFeatureFunction { + public: + MockStatelessFeatureFunction(const string& desc, size_t n) : + StatelessFeatureFunction(desc,n) {} + virtual void Evaluate(const PhraseBasedFeatureContext&, ScoreComponentCollection*) const {} + virtual void EvaluateChart(const ChartBasedFeatureContext&, ScoreComponentCollection*) const {} +}; + +class MockSingleFeature : public MockStatelessFeatureFunction { + public: + MockSingleFeature(): MockStatelessFeatureFunction("MockSingle",1) {} + std::string GetScoreProducerWeightShortName(unsigned) const {return "sf";} +}; + +class MockMultiFeature : public MockStatelessFeatureFunction { + public: + MockMultiFeature(): MockStatelessFeatureFunction("MockMulti", 5) {} + std::string GetScoreProducerWeightShortName(unsigned) const {return "mf";} +}; + +class MockSparseFeature : public MockStatelessFeatureFunction { + public: + MockSparseFeature(): MockStatelessFeatureFunction("MockSparse", ScoreProducer::unlimited) {} + std::string GetScoreProducerWeightShortName(unsigned) const {return "sf";} +}; + + +struct MockProducers { + MockProducers() {} + + MockSingleFeature single; + MockMultiFeature multi; + MockSparseFeature sparse; +}; + +BOOST_FIXTURE_TEST_CASE(ctor, MockProducers) +{ + ScoreComponentCollection scc; + BOOST_CHECK_EQUAL(scc.GetScoreForProducer(&single),0); + float expected[] = {0,0,0,0,0}; + std::vector<float> actual= scc.GetScoresForProducer(&multi); + BOOST_CHECK_EQUAL_COLLECTIONS(expected, expected+5, actual.begin(), actual.begin()+5); +} + +BOOST_FIXTURE_TEST_CASE(plusequals, MockProducers) +{ + float arr1[] = {1,2,3,4,5}; + float arr2[] = {2,4,6,8,10}; + std::vector<float> vec1(arr1,arr1+5); + std::vector<float> vec2(arr2,arr2+5); + + ScoreComponentCollection scc; + scc.PlusEquals(&single, 3.4f); + BOOST_CHECK_EQUAL(scc.GetScoreForProducer(&single), 3.4f); + scc.PlusEquals(&multi,vec1); + std::vector<float> actual = scc.GetScoresForProducer(&multi); + BOOST_CHECK_EQUAL_COLLECTIONS(vec1.begin(),vec1.end() + ,actual.begin(), actual.end()); + scc.PlusEquals(&multi,vec1); + actual = scc.GetScoresForProducer(&multi); + BOOST_CHECK_EQUAL_COLLECTIONS(vec2.begin(),vec2.end(), + actual.begin(), actual.end()); + + BOOST_CHECK_EQUAL(scc.GetScoreForProducer(&single), 3.4f); +} + +BOOST_FIXTURE_TEST_CASE(sparse_feature, MockProducers) +{ + ScoreComponentCollection scc; + scc.Assign(&sparse, "first", 1.3f); + scc.Assign(&sparse, "second", 2.1f); + BOOST_CHECK_EQUAL( scc.GetScoreForProducer(&sparse,"first"), 1.3f); + BOOST_CHECK_EQUAL( scc.GetScoreForProducer(&sparse,"second"), 2.1f); + BOOST_CHECK_EQUAL( scc.GetScoreForProducer(&sparse,"third"), 0.0f); + scc.Assign(&sparse, "first", -1.9f); + BOOST_CHECK_EQUAL( scc.GetScoreForProducer(&sparse,"first"), -1.9f); + scc.PlusEquals(&sparse, "first", -1.9f); + BOOST_CHECK_EQUAL( scc.GetScoreForProducer(&sparse,"first"), -3.8f); +} + +/* + Doesn't work because of the static registration of ScoreProducers + in ScoreComponentCollection. +BOOST_FIXTURE_TEST_CASE(save, MockProducers) +{ + ScoreComponentCollection scc; + scc.Assign(&sparse, "first", 1.1f); + scc.Assign(&single, 0.25f); + float arr[] = {1,2.1,3,4,5}; + std::vector<float> vec1(arr,arr+5); + scc.Assign(&multi,vec1); + ostringstream out; + scc.Save(out); + cerr << out.str() << endl; + istringstream in (out.str()); + string line; + getline(in,line); + BOOST_CHECK_EQUAL(line, "MockSingle:4_1 0.25"); + getline(in,line); + BOOST_CHECK_EQUAL(line, "MockMulti:4_1 1"); + getline(in,line); + BOOST_CHECK_EQUAL(line, "MockMulti:4_2 2.1"); + getline(in,line); + BOOST_CHECK_EQUAL(line, "MockMulti:4_3 3"); + getline(in,line); + BOOST_CHECK_EQUAL(line, "MockMulti:4_4 4"); + getline(in,line); + BOOST_CHECK_EQUAL(line, "MockMulti:4_5 5"); + getline(in,line); + BOOST_CHECK_EQUAL(line,"MockSparse:4_first 1.1"); + BOOST_CHECK(!getline(in,line)); +} +*/ + + +BOOST_AUTO_TEST_SUITE_END() + |