dedent

langgenius · May 1, 2024 · b603185 · b603185
1 parent a30e442
commit b603185
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 32 deletions.
diff --git a/api/tests/unit_tests/utils/position_helper/test_position_helper.py b/api/tests/unit_tests/utils/position_helper/test_position_helper.py
@@ -1,3 +1,5 @@
+from textwrap import dedent
+
 import pytest
 
 from core.utils.position_helper import get_position_map
@@ -6,15 +8,16 @@
 @pytest.fixture
 def prepare_example_positions_yaml(tmp_path, monkeypatch) -> str:
     monkeypatch.chdir(tmp_path)
-    tmp_path.joinpath("example_positions.yaml").write_text("""
-- first
-- second
-# - commented
-- third
-
-- 9999999999999
-- forth
-            """)
+    tmp_path.joinpath("example_positions.yaml").write_text(dedent(
+        """\
+        - first
+        - second
+        # - commented
+        - third
+        
+        - 9999999999999
+        - forth
+        """))
     return str(tmp_path)
 
 

diff --git a/api/tests/unit_tests/utils/yaml/test_yaml_utils.py b/api/tests/unit_tests/utils/yaml/test_yaml_utils.py
@@ -1,3 +1,5 @@
+from textwrap import dedent
+
 import pytest
 from yaml import YAMLError
 
@@ -12,36 +14,38 @@
 def prepare_example_yaml_file(tmp_path, monkeypatch) -> str:
     monkeypatch.chdir(tmp_path)
     file_path = tmp_path.joinpath(EXAMPLE_YAML_FILE)
-    file_path.write_text("""
-address:
-    city: Example City
-    country: Example Country
-age: 30
-gender: male
-languages:
-    - Python
-    - Java
-    - C++
-empty_key:
-      """)
+    file_path.write_text(dedent(
+        """\
+        address:
+            city: Example City
+            country: Example Country
+        age: 30
+        gender: male
+        languages:
+            - Python
+            - Java
+            - C++
+        empty_key:
+        """))
     return str(file_path)
 
 
 @pytest.fixture
 def prepare_invalid_yaml_file(tmp_path, monkeypatch) -> str:
     monkeypatch.chdir(tmp_path)
     file_path = tmp_path.joinpath(INVALID_YAML_FILE)
-    file_path.write_text("""
-address:
-           city: Example City
-    country: Example Country
-age: 30
-gender: male
-languages:
-- Python
-- Java
-- C++
-      """)
+    file_path.write_text(dedent(
+        """\
+        address:
+                   city: Example City
+            country: Example Country
+        age: 30
+        gender: male
+        languages:
+        - Python
+        - Java
+        - C++
+        """))
     return str(file_path)