Code Review
/
csit.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
review
|
tree
raw
|
inline
| side by side
Ansible: Increase TG hugepages
[csit.git]
/
resources
/
tools
/
wrk
/
wrk_traffic_profile_parser.py
diff --git
a/resources/tools/wrk/wrk_traffic_profile_parser.py
b/resources/tools/wrk/wrk_traffic_profile_parser.py
index
1994b61
..
f553465
100644
(file)
--- a/
resources/tools/wrk/wrk_traffic_profile_parser.py
+++ b/
resources/tools/wrk/wrk_traffic_profile_parser.py
@@
-20,7
+20,7
@@
See LLD for the structure of a wrk traffic profile.
from os.path import isfile
from pprint import pformat
from os.path import isfile
from pprint import pformat
-from yaml import load, YAMLError
+from yaml import
safe_
load, YAMLError
from robot.api import logger
from resources.tools.wrk.wrk_errors import WrkError
from robot.api import logger
from resources.tools.wrk.wrk_errors import WrkError
@@
-61,8
+61,8
@@
class WrkTrafficProfile:
self.profile_name = profile_name
try:
self.profile_name = profile_name
try:
- with open(self.profile_name,
"r
") as profile_file:
- self.traffic_profile = load(profile_file)
+ with open(self.profile_name,
u"rt
") as profile_file:
+ self.traffic_profile =
safe_
load(profile_file)
except IOError as err:
raise WrkError(
msg=f"An error occurred while opening the file "
except IOError as err:
raise WrkError(
msg=f"An error occurred while opening the file "