X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/bd1e98034907463f5d86f4d87e89202dc6c34dd4..7f7673d941a947a8d392c8c0866d3d588affc174:/tests/test_format.py?ds=inline diff --git a/tests/test_format.py b/tests/test_format.py index a995bd3..1916146 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", @@ -71,6 +72,11 @@ PY310_CASES: List[str] = [ "parenthesized_context_managers", ] +PY311_CASES: List[str] = [ + "pep_654", + "pep_654_style", +] + PREVIEW_CASES: List[str] = [ # string processing "cantfit", @@ -83,6 +89,8 @@ PREVIEW_CASES: List[str] = [ "remove_except_parens", "remove_for_brackets", "one_element_subscript", + "remove_await_parens", + "return_annotation_brackets", ] SOURCES: List[str] = [ @@ -192,6 +200,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) @@ -214,6 +232,13 @@ def test_patma_invalid() -> None: exc_info.match("Cannot parse: 10:11") +@pytest.mark.parametrize("filename", PY311_CASES) +def test_python_311(filename: str) -> None: + source, expected = read_data(filename) + mode = black.Mode(target_versions={black.TargetVersion.PY311}) + assert_format(source, expected, mode, minimum_version=(3, 11)) + + def test_python_2_hint() -> None: with pytest.raises(black.parsing.InvalidInput) as exc_info: assert_format("print 'daylily'", "print 'daylily'")