diff options
| author | Ang Li <angli@google.com> | 2019-05-22 10:51:26 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2019-05-22 10:51:26 -0700 |
| commit | 2ab5eee6f211077a16b0e1c8a4ed1a8adcb90d47 (patch) | |
| tree | da83b4681d4bc487143c7991c437976d7b08f77a /tests/mobly/records_test.py | |
| parent | fc844080ab152a1b1853197d62ecf0798b209002 (diff) | |
| download | mobly-2ab5eee6f211077a16b0e1c8a4ed1a8adcb90d47.tar.gz | |
Fix yaml load usage warnings. (#578)
Diffstat (limited to 'tests/mobly/records_test.py')
| -rwxr-xr-x | tests/mobly/records_test.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/mobly/records_test.py b/tests/mobly/records_test.py index ba25a69..a374347 100755 --- a/tests/mobly/records_test.py +++ b/tests/mobly/records_test.py @@ -361,7 +361,7 @@ class RecordsTest(unittest.TestCase): writer = records.TestSummaryWriter(dump_path) writer.dump(record1.to_dict(), records.TestSummaryEntryType.RECORD) with io.open(dump_path, 'r', encoding='utf-8') as f: - content = yaml.load(f) + content = yaml.safe_load(f) self.assertEqual(content['Type'], records.TestSummaryEntryType.RECORD.value) self.assertEqual(content[records.TestResultEnums.RECORD_DETAILS], @@ -380,7 +380,7 @@ class RecordsTest(unittest.TestCase): writer = records.TestSummaryWriter(dump_path) writer.dump(record1.to_dict(), records.TestSummaryEntryType.RECORD) with io.open(dump_path, 'r', encoding='utf-8') as f: - content = yaml.load(f) + content = yaml.safe_load(f) self.assertEqual(content['Type'], records.TestSummaryEntryType.RECORD.value) self.assertEqual(content[records.TestResultEnums.RECORD_DETAILS], @@ -398,7 +398,7 @@ class RecordsTest(unittest.TestCase): writer.dump(data, records.TestSummaryEntryType.USER_DATA) with io.open(dump_path, 'r', encoding='utf-8') as f: contents = [] - for c in yaml.load_all(f): + for c in yaml.safe_load_all(f): contents.append(c) for content in contents: self.assertEqual(content['Type'], |