Use parentheses on method access on float and int literals (#2799)

[etc/vim.git] / tests / test_format.py

diff --git a/tests/test_format.py b/tests/test_format.py
index 3895a095e863c2ec503fa43e272add8062da9f8e..aef22545f5b4bd128469ed4ddbb51d47832036c3 100644 (file)
--- a/tests/test_format.py
+++ b/tests/test_format.py
@@ -15,6 +15,7 @@ from tests.util import (
 )
 
 SIMPLE_CASES: List[str] = [
 )
 
 SIMPLE_CASES: List[str] = [

+    "attribute_access_on_number_literals",

     "beginning_backslash",
     "bracketmatch",
     "class_blank_parentheses",
     "beginning_backslash",
     "bracketmatch",
     "class_blank_parentheses",


@@ -48,9 +49,13 @@ SIMPLE_CASES: List[str] = [
     "function2",
     "function_trailing_comma",
     "import_spacing",
     "function2",
     "function_trailing_comma",
     "import_spacing",

+    "power_op_spacing",

     "remove_parens",
     "slices",
     "string_prefixes",
     "remove_parens",
     "slices",
     "string_prefixes",

+    "trailing_comma_optional_parens1",
+    "trailing_comma_optional_parens2",
+    "trailing_comma_optional_parens3",

     "tricky_unicode_symbols",
     "tupleassign",
 ]
     "tricky_unicode_symbols",
     "tupleassign",
 ]


@@ -252,3 +257,9 @@ def test_python38() -> None:
 def test_python39() -> None:
     source, expected = read_data("python39")
     assert_format(source, expected, minimum_version=(3, 9))
 def test_python39() -> None:
     source, expected = read_data("python39")
     assert_format(source, expected, minimum_version=(3, 9))

+
+
+def test_power_op_newline() -> None:
+    # requires line_length=0
+    source, expected = read_data("power_op_newline")
+    assert_format(source, expected, mode=black.Mode(line_length=0))