Belle II Software  release-08-01-10
test_density_scaler.py
1 
8 
9 '''
10 Test if the DensityScaler module works.
11 N.B.: this test can not be run in a light-release enviroment.
12 '''
13 
14 import basf2 as b2
15 import b2test_utils as b2tu
16 
17 b2tu.skip_test_if_light()
18 
19 import generators as ge # noqa
20 import simulation as si # noqa
21 
22 
23 main = b2.Path()
24 
25 main.add_module('EventInfoSetter')
26 
27 main.add_module('Progress')
28 
29 ge.add_evtgen_generator(main, 'mixed')
30 
31 # This is the relevant piece of this test.
32 main.add_module('DensityScaler', scaling={'BeamPipe': 0.85, 'ECL': 1.5})
33 
34 si.add_simulation(main)
35 
36 b2.process(main)