pjht/buku
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #406 from rachmadaniHaryono/feature/yaml-load

Browse Source 
yaml load
This commit is contained in:
Mischievous Meerkat 2019-07-21 10:26:23 +05:30 committed by GitHub
commit c1dfe17997
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_bukuDb.py
View File

@ -1318,7 +1318,10 @@ def test_load_chrome_database(chrome_db, add_pt):
dump_data = False # NOTE: change this value to dump data dump_data = False # NOTE: change this value to dump data
if not dump_data: if not dump_data:
with open(res_yaml_file, 'r') as f: with open(res_yaml_file, 'r') as f:
res_yaml = yaml.load(f, Loader=yaml.FullLoader) try:
res_yaml = yaml.load(f, Loader=yaml.FullLoader)
except RuntimeError:
res_yaml = yaml.safe_load(f)
# init # init
import buku import buku
bdb = buku.BukuDb() bdb = buku.BukuDb()
@ -1359,7 +1362,10 @@ def test_load_firefox_database(firefox_db, add_pt):
res_yaml_file = firefox_db[1] if add_pt else firefox_db[2] res_yaml_file = firefox_db[1] if add_pt else firefox_db[2]
if not dump_data: if not dump_data:
with open(res_yaml_file, 'r') as f: with open(res_yaml_file, 'r') as f:
res_yaml = yaml.load(f) try:
res_yaml = yaml.load(f)
except RuntimeError:
res_yaml = yaml.safe_load(f)
# init # init
import buku import buku
bdb = buku.BukuDb() bdb = buku.BukuDb()