chg: dev: use safe load only

This commit is contained in:
rachmadaniHaryono 2021-05-01 23:17:24 +08:00
parent 58adeb78bf
commit 03d808072f

View File

@ -1327,10 +1327,7 @@ def test_load_firefox_database(firefox_db, add_pt):
res_yaml_file = firefox_db[1] if add_pt else firefox_db[2]
if not dump_data:
with open(res_yaml_file, "r") as f:
try:
res_yaml = yaml.load(f)
except RuntimeError:
res_yaml = yaml.load(f, Loader=PrettySafeLoader)
res_yaml = yaml.load(f, Loader=PrettySafeLoader)
# init
import buku