Tools/test/TestVertexReader.h

#include "cute/cute.h"
#include "cute/cute_suite.h"

#include <boost/shared_ptr.hpp>

#include "../interface/Readers/VertexReader.h"
#include "TChain.h"

using namespace BAT;

struct TestVertexReader {
        boost::shared_ptr<TChain> input;
        boost::shared_ptr<VertexReader> reader;
        VertexPointer vertex;

        TestVertexReader() :
                input(new TChain(NTupleEventReader::EVENT_CHAIN)),
                reader(new VertexReader(input)),
                vertex() {
                input->Add(InputFile::ttbar);
                input->SetBranchStatus("*", 0);
                reader->initialise();
                input->GetEntry(1);
                vertex = reader->getVertices().front();
        }

        void testVertexZPosition() {
                ASSERT_EQUAL_DELTA(1.86712, vertex->absoluteZPosition(), 0.00001);
        }

        void testVertexRho(){
                ASSERT_EQUAL_DELTA(0.461491, vertex->absoluteRho(), 0.000001);
        }

        void testVertexIsFake(){
                ASSERT(vertex->isFake() == false);
        }

        void testVertexNDOF(){
                ASSERT_EQUAL(111, vertex->ndof());
        }
};

cute::suite make_suite_TestVertexReader() {
        cute::suite s;

        s.push_back(CUTE_SMEMFUN(TestVertexReader, testVertexZPosition));
        s.push_back(CUTE_SMEMFUN(TestVertexReader, testVertexRho));
        s.push_back(CUTE_SMEMFUN(TestVertexReader, testVertexIsFake));
        s.push_back(CUTE_SMEMFUN(TestVertexReader, testVertexNDOF));

        return s;
}
Revision:	1.1.1.1 (vendor branch)
Committed:	Thu Dec 1 16:28:47 2011 UTC (13 years, 5 months ago) by dhidas
Content type:	text/plain
Branch:	dhidas, MAIN
CVS Tags:	START, HEAD
Changes since 1.1:	+0 -0 lines
Error occurred while calculating annotation data.
Log Message:	osu copy modified
#	Content
1	#include "cute/cute.h"
2	#include "cute/cute_suite.h"
3
4	#include <boost/shared_ptr.hpp>
5
6	#include "../interface/Readers/VertexReader.h"
7	#include "TChain.h"
8
9	using namespace BAT;
10
11	struct TestVertexReader {
12	boost::shared_ptr<TChain> input;
13	boost::shared_ptr<VertexReader> reader;
14	VertexPointer vertex;
15
16	TestVertexReader() :
17	input(new TChain(NTupleEventReader::EVENT_CHAIN)),
18	reader(new VertexReader(input)),
19	vertex() {
20	input->Add(InputFile::ttbar);
21	input->SetBranchStatus("*", 0);
22	reader->initialise();
23	input->GetEntry(1);
24	vertex = reader->getVertices().front();
25	}
26
27	void testVertexZPosition() {
28	ASSERT_EQUAL_DELTA(1.86712, vertex->absoluteZPosition(), 0.00001);
29	}
30
31	void testVertexRho(){
32	ASSERT_EQUAL_DELTA(0.461491, vertex->absoluteRho(), 0.000001);
33	}
34
35	void testVertexIsFake(){
36	ASSERT(vertex->isFake() == false);
37	}
38
39	void testVertexNDOF(){
40	ASSERT_EQUAL(111, vertex->ndof());
41	}
42	};
43
44	cute::suite make_suite_TestVertexReader() {
45	cute::suite s;
46
47	s.push_back(CUTE_SMEMFUN(TestVertexReader, testVertexZPosition));
48	s.push_back(CUTE_SMEMFUN(TestVertexReader, testVertexRho));
49	s.push_back(CUTE_SMEMFUN(TestVertexReader, testVertexIsFake));
50	s.push_back(CUTE_SMEMFUN(TestVertexReader, testVertexNDOF));
51
52	return s;
53	}