admin/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

chore(api/tests): apply ruff reformat #7590 (#7591)

Browse Source 
Co-authored-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
Bowen Liang
2024-08-23 23:52:25 +08:00
committed by GitHub
parent 2da63654e5
commit b035c02f78
155 changed files with 4279 additions and 5925 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
api/tests/unit_tests/utils/position_helper/test_position_helper.py
View File

@@ -8,8 +8,9 @@ from core.helper.position_helper import get_position_map, is_filtered, pin_posit
@pytest.fixture
def prepare_example_positions_yaml(tmp_path, monkeypatch) -> str:
monkeypatch.chdir(tmp_path)
tmp_path.joinpath("example_positions.yaml").write_text(dedent(
"""\
tmp_path.joinpath("example_positions.yaml").write_text(
dedent(
"""\
- first
- second
# - commented
@@ -17,57 +18,54 @@ def prepare_example_positions_yaml(tmp_path, monkeypatch) -> str:
- 9999999999999
- forth
"""))
"""
)
)
return str(tmp_path)
@pytest.fixture
def prepare_empty_commented_positions_yaml(tmp_path, monkeypatch) -> str:
monkeypatch.chdir(tmp_path)
tmp_path.joinpath("example_positions_all_commented.yaml").write_text(dedent(
"""\
tmp_path.joinpath("example_positions_all_commented.yaml").write_text(
dedent(
"""\
# - commented1
# - commented2
-
-
"""))
"""
)
)
return str(tmp_path)
def test_position_helper(prepare_example_positions_yaml):
position_map = get_position_map(
folder_path=prepare_example_positions_yaml,
file_name='example_positions.yaml')
position_map = get_position_map(folder_path=prepare_example_positions_yaml, file_name="example_positions.yaml")
assert len(position_map) == 4
assert position_map == {
'first': 0,
'second': 1,
'third': 2,
'forth': 3,
"first": 0,
"second": 1,
"third": 2,
"forth": 3,
}
def test_position_helper_with_all_commented(prepare_empty_commented_positions_yaml):
position_map = get_position_map(
folder_path=prepare_empty_commented_positions_yaml,
file_name='example_positions_all_commented.yaml')
folder_path=prepare_empty_commented_positions_yaml, file_name="example_positions_all_commented.yaml"
)
assert position_map == {}
def test_excluded_position_data(prepare_example_positions_yaml):
position_map = get_position_map(
folder_path=prepare_example_positions_yaml,
file_name='example_positions.yaml'
)
pin_list = ['forth', 'first']
position_map = get_position_map(folder_path=prepare_example_positions_yaml, file_name="example_positions.yaml")
pin_list = ["forth", "first"]
include_set = set()
exclude_set = {'9999999999999'}
exclude_set = {"9999999999999"}
position_map = pin_position_map(
original_position_map=position_map,
pin_list=pin_list
)
position_map = pin_position_map(original_position_map=position_map, pin_list=pin_list)
data = [
"forth",
@@ -90,22 +88,16 @@ def test_excluded_position_data(prepare_example_positions_yaml):
)
# assert the result in the correct order
assert sorted_data == ['forth', 'first', 'second', 'third', 'extra1', 'extra2']
assert sorted_data == ["forth", "first", "second", "third", "extra1", "extra2"]
def test_included_position_data(prepare_example_positions_yaml):
position_map = get_position_map(
folder_path=prepare_example_positions_yaml,
file_name='example_positions.yaml'
)
pin_list = ['forth', 'first']
include_set = {'forth', 'first'}
position_map = get_position_map(folder_path=prepare_example_positions_yaml, file_name="example_positions.yaml")
pin_list = ["forth", "first"]
include_set = {"forth", "first"}
exclude_set = {}
position_map = pin_position_map(
original_position_map=position_map,
pin_list=pin_list
)
position_map = pin_position_map(original_position_map=position_map, pin_list=pin_list)
data = [
"forth",
@@ -128,4 +120,4 @@ def test_included_position_data(prepare_example_positions_yaml):
)
# assert the result in the correct order
assert sorted_data == ['forth', 'first']
assert sorted_data == ["forth", "first"]