update

langgenius · Apr 30, 2024 · 536bc52 · 536bc52
1 parent 3927dd1
commit 536bc52
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/api/tests/unit_tests/utils/yaml/test_yaml_utils.py b/api/tests/unit_tests/utils/yaml/test_yaml_utils.py
@@ -35,6 +35,7 @@ def prepare_invalid_yaml_file(tmp_path, monkeypatch) -> str:
            city: Example City
     country: Example Country
 age: 30
+empty_key:
 gender: male
 languages:
 - Python
@@ -57,6 +58,7 @@ def test_load_valid_yaml_file(prepare_example_yaml_file):
     assert len(yaml_data) > 0
     assert yaml_data['age'] == 30
     assert yaml_data['gender'] == 'male'
+    assert yaml_data.get('empty_key') is None
     assert yaml_data['address']['city'] == 'Example City'
     assert set(yaml_data['languages']) == {'Python', 'Java', 'C++'}