X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/a57ab326b20b720518ab6f513bd0f8ba357d8d86..431bd09e15247431056894bd6444dee7c22893f0:/tests/test_format.py diff --git a/tests/test_format.py b/tests/test_format.py index 269bbac..fd5f596 100644 --- a/tests/test_format.py +++ b/tests/test_format.py @@ -44,6 +44,7 @@ SIMPLE_CASES: List[str] = [ "fmtskip4", "fmtskip5", "fmtskip6", + "fmtskip7", "fstring", "function", "function2", @@ -75,10 +76,16 @@ PREVIEW_CASES: List[str] = [ # string processing "cantfit", "comments7", + "comments8", "long_strings", "long_strings__edge_case", "long_strings__regression", "percent_precedence", + "remove_except_parens", + "remove_for_brackets", + "one_element_subscript", + "remove_await_parens", + "return_annotation_brackets", ] SOURCES: List[str] = [ @@ -188,6 +195,16 @@ def test_pep_570() -> None: assert_format(source, expected, minimum_version=(3, 8)) +def test_remove_with_brackets() -> None: + source, expected = read_data("remove_with_brackets") + assert_format( + source, + expected, + black.Mode(preview=True), + minimum_version=(3, 9), + ) + + @pytest.mark.parametrize("filename", PY310_CASES) def test_python_310(filename: str) -> None: source, expected = read_data(filename)