[MBI] Import generator_utils as gen.

[simgrid.git] / teshsuite / smpi / MBI / ResleakGenerator.py

diff --git a/teshsuite/smpi/MBI/ResleakGenerator.py b/teshsuite/smpi/MBI/ResleakGenerator.py
index e83205f..d1d84ea 100755 (executable)
--- a/teshsuite/smpi/MBI/ResleakGenerator.py
+++ b/teshsuite/smpi/MBI/ResleakGenerator.py
@@ -5,7 +5,7 @@
 
 import os
 import sys
-from generator_utils import *
+import generator_utils as gen
 
 template = """// @{generatedby}@
 /* ///////////////////////// The MPI Bugs Initiative ////////////////////////
@@ -50,7 +50,7 @@ static void myOp(int *invec, int *inoutvec, int *len, MPI_Datatype *dtype) {
 int main(int argc, char **argv) {
   int nprocs = -1;
   int rank = -1;
-  int i=1, j=1, size=1;
+  int i=1, j=0, size=1;
   int color =0;
 
   MPI_Init(&argc, &argv);
@@ -63,11 +63,11 @@ int main(int argc, char **argv) {
 
   @{change_size}@
   @{init}@
-  @{loop}@  
+  @{loop}@
   @{operation}@
-  @{cond}@  
+  @{cond}@
   @{fini}@
-  @{end}@  
+  @{end}@
 
   @{free}@
 
@@ -78,16 +78,16 @@ int main(int argc, char **argv) {
 """
 
 # Generate code with one collective
-for call in tcoll:
+for call in gen.tcoll:
     patterns = {}
     patterns = {'call': call}
     patterns['generatedby'] = f'DO NOT EDIT: this file was generated by {os.path.basename(sys.argv[0])}. DO NOT EDIT.'
     patterns['toolfeature'] = 'Yes'
     patterns['call'] = call
-    patterns['operation'] = operation[call]("1")
-    patterns['init'] = init[call]("1")
-    patterns['fini'] = fini[call]("1")
-    patterns['free'] = free[call]("1")
+    patterns['operation'] = gen.operation[call]("1")
+    patterns['init'] = gen.init[call]("1")
+    patterns['fini'] = gen.fini[call]("1")
+    patterns['free'] = gen.free[call]("1")
     missing = patterns['fini']
     patterns['loop'] = ''
     patterns['cond'] = ''
@@ -100,26 +100,26 @@ for call in tcoll:
     replace['longdesc'] = f'{call} correctly used'
     replace['outcome'] = 'OK'
     replace['errormsg'] = ''
-    make_file(template, f'ResLeak_{call}_ok.c', replace)
+    gen.make_file(template, f'ResLeak_{call}_ok.c', replace)
 
     # Generate the resleak
     replace = patterns
     replace['shortdesc'] = '@{call}@ has no free'
     replace['longdesc'] = '@{call}@ has no free'
-    replace['outcome'] = f'ERROR: {error[call]}'
+    replace['outcome'] = f'ERROR: {gen.error[call]}'
     replace['errormsg'] = 'Resleak. @{call}@ at @{filename}@:@{line:MBIERROR}@ has no free.'
     replace['fini'] = ' /* MBIERROR MISSING: ' + missing + ' */'
-    make_file(template, f'ResLeak_{call}_nok.c', replace)
+    gen.make_file(template, f'ResLeak_{call}_nok.c', replace)
 
     # Generate multiple resleak
     replace = patterns
     replace['shortdesc'] = '@{call}@ lacks several free'
     replace['longdesc'] = '@{call}@ lacks several free'
-    replace['outcome'] = f'ERROR: {error[call]}'
+    replace['outcome'] = f'ERROR: {gen.error[call]}'
     replace['errormsg'] = 'Resleak. @{call}@ at @{filename}@:@{line:MBIERROR}@ lacks several free.'
     replace['change_size'] = 'size=PARAM_PER_ITERATION;'
     replace['loop'] = 'for (i = 0; i < ITERATIONS; i++) {\n    for (j = 0; j < PARAM_PER_ITERATION; j++) {'
     replace['cond'] = '      if (j < PARAM_PER_ITERATION - PARAM_LOST_PER_ITERATION) {'
-    replace['fini'] = fini[call]("1") + ' /* MBIERROR */'
-    replace['end'] = '      }\n                }\n     }'
-    make_file(template, f'ResLeak_multiple_{call}_nok.c', replace)
+    replace['fini'] = gen.fini[call]("1") + ' /* MBIERROR */'
+    replace['end'] = '      }\n     }\n   }'
+    gen.make_file(template, f'ResLeak_multiple_{call}_nok.c', replace)