Append 3 more punt path protection TCs

[integration/test.git] / csit / scripts / data_generate.py

diff --git a/csit/scripts/data_generate.py b/csit/scripts/data_generate.py
index 9a7f714aae2b9ea7785124375ad130c67b9546fe..b0725210888f6d88d1dec40f3a898e3f5a70ed36 100644 (file)
--- a/csit/scripts/data_generate.py
+++ b/csit/scripts/data_generate.py
@@ -34,9 +34,8 @@ def generate():
                 lines = file.readlines()
             props = lines[0].strip().split(',')
             vals = lines[1].strip().split(',')
-            BODY['plots'][key][props[0]] = float(vals[0])
-            BODY['plots'][key][props[1]] = float(vals[1])
-            BODY['plots'][key][props[2]] = float(vals[2])
+            for i in range(len(props)):
+                BODY['plots'][key][props[i]] = float(vals[i])
 
     # Fill the required parameters whose values are obtained from environment.
 
@@ -61,9 +60,4 @@ def generate():
         - datetime.strptime(starttime, '%Y%m%d %H:%M:%S.%f')
     BODY['duration'] = str(elap_time)
 
-    BODY = {
-        'type': BODY['test-type'],
-        BODY['test-type']: BODY
-    }
-
     return BODY