Add new tests for format feature

psf · Jan 23, 2022 · 5ee6627 · 5ee6627
1 parent e65f935
commit 5ee6627
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/tests/data/use_braces_in_int_method_calls.py b/tests/data/use_braces_in_int_method_calls.py
@@ -0,0 +1,24 @@
+count = 5 .bit_count()
+abs = 10 .__abs__()
+is_integer = 10.5 .is_integer()
+
+if 10 .real: ...
+
+for number in range(1, 10):
+    if number.imag: ...
+
+
+# output
+count = (5).bit_count()
+abs = (10).__abs__()
+is_integer = (10.5).is_integer()
+
+if (10).real:
+    ...
+
+for number in range(1, 10):
+    if number.imag:
+        ...
+
+
+
diff --git a/tests/test_format.py b/tests/test_format.py
@@ -53,6 +53,7 @@
     "string_prefixes",
     "tricky_unicode_symbols",
     "tupleassign",
+    "use_braces_in_int_method_calls",
 ]
 
 PY310_CASES: List[str] = [