@@ -148,11 +148,12 @@ def test_simd_cpp_eemumua_float(self):
148
148
""" % self .run_dir
149
149
150
150
open (pjoin (self .path , 'mg5_cmd' ),'w' ).write (cmd )
151
-
152
- subprocess .call ([sys .executable , pjoin (MG5DIR , 'bin' ,'mg5_aMC' ),
153
- pjoin (self .path , 'mg5_cmd' )],
151
+ newenv = os .environ .copy ()
152
+ newenv ["PYTHONPATH" ] = f"..:{ newenv ['PYTHONPATH' ]} "
153
+ subprocess .call ([sys .executable , pjoin (MG5DIR , 'bin' ,'mg5_aMC' ,'-m' ,'CUDACPP_OUTPUT' ),
154
+ pjoin (self .path , 'mg5_cmd' )], env = newenv ,
154
155
#cwd=self.path,
155
- stdout = stdout , stderr = stderr )
156
+ stdout = stdout , stderr = stderr )
156
157
157
158
self .check_parton_output (cross = 0.0266 , error = 0.0002854 , run_name = 'run_01' , html = True )
158
159
event = '%s/Events/run_01/unweighted_events.lhe' % self .run_dir
@@ -197,11 +198,12 @@ def test_simd_cpp_heft_ggh_double(self):
197
198
""" % self .run_dir
198
199
199
200
open (pjoin (self .path , 'mg5_cmd' ),'w' ).write (cmd )
200
-
201
- subprocess .call ([sys .executable , pjoin (MG5DIR , 'bin' ,'mg5_aMC' ),
202
- pjoin (self .path , 'mg5_cmd' )],
201
+ newenv = os .environ .copy ()
202
+ newenv ["PYTHONPATH" ] = f"..:{ newenv ['PYTHONPATH' ]} "
203
+ subprocess .call ([sys .executable , pjoin (MG5DIR , 'bin' ,'mg5_aMC' ,'-m' ,'CUDACPP_OUTPUT' ),
204
+ pjoin (self .path , 'mg5_cmd' )], env = newenv ,
203
205
#cwd=self.path,
204
- stdout = stdout , stderr = stderr )
206
+ stdout = stdout , stderr = stderr )
205
207
206
208
self .check_parton_output (cross = 0.01859 , error = 0.0002853789088650386 , run_name = 'run_01' , html = True )
207
209
event = '%s/Events/run_01/unweighted_events.lhe' % self .run_dir
@@ -246,11 +248,12 @@ def test_simd_cpp_pptt_mixed(self):
246
248
""" % self .run_dir
247
249
248
250
open (pjoin (self .path , 'mg5_cmd' ),'w' ).write (cmd )
249
-
250
- subprocess .call ([sys .executable , pjoin (MG5DIR , 'bin' ,'mg5_aMC' ),
251
- pjoin (self .path , 'mg5_cmd' )],
251
+ newenv = os .environ .copy ()
252
+ newenv ["PYTHONPATH" ] = f"..:{ newenv ['PYTHONPATH' ]} "
253
+ subprocess .call ([sys .executable , pjoin (MG5DIR , 'bin' ,'mg5_aMC' ,'-m' ,'CUDACPP_OUTPUT' ),
254
+ pjoin (self .path , 'mg5_cmd' )], env = newenv ,
252
255
#cwd=self.path,
253
- stdout = stdout , stderr = stderr )
256
+ stdout = stdout , stderr = stderr )
254
257
255
258
self .check_parton_output (cross = 505.5 , error = 2.749 , run_name = 'run_01' , html = True )
256
259
event = '%s/Events/run_01/unweighted_events.lhe' % self .run_dir
@@ -302,11 +305,12 @@ def test_simd_cpp_vector_size(self):
302
305
""" % self .run_dir
303
306
304
307
open (pjoin (self .path , 'mg5_cmd' ),'w' ).write (cmd )
305
-
306
- subprocess .call ([sys .executable , pjoin (MG5DIR , 'bin' ,'mg5_aMC' ),
307
- pjoin (self .path , 'mg5_cmd' )],
308
+ newenv = os .environ .copy ()
309
+ newenv ["PYTHONPATH" ] = f"..:{ newenv ['PYTHONPATH' ]} "
310
+ subprocess .call ([sys .executable , pjoin (MG5DIR , 'bin' ,'mg5_aMC' ,'-m' ,'CUDACPP_OUTPUT' ),
311
+ pjoin (self .path , 'mg5_cmd' )], env = newenv ,
308
312
#cwd=self.path,
309
- stdout = stdout , stderr = stderr )
313
+ stdout = stdout , stderr = stderr )
310
314
311
315
self .check_parton_output (cross = 505.5 , error = 2.749 , run_name = 'run_01' , html = True )
312
316
event = '%s/Events/run_01/unweighted_events.lhe' % self .run_dir
0 commit comments