fix bug when rewriting deprecated unittest functions

asottile · Sep 21, 2023 · cf3053a · cf3053a
1 parent 1231861
commit cf3053a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/pyupgrade/_plugins/unittest_aliases.py b/pyupgrade/_plugins/unittest_aliases.py
@@ -57,13 +57,13 @@ def visit_Call(
         )
         yield ast_to_offset(node.func), func
     elif (
-        isinstance(node.func, ast.Attribute) and
-        isinstance(node.func.value, ast.Name) and
-        node.func.value.id == 'unittest' and
-        node.func.attr in FUNCTION_MAPPING and
-        not has_starargs(node) and
-        not node.keywords and
-        len(node.args) == 1
+            isinstance(node.func, ast.Attribute) and
+            isinstance(node.func.value, ast.Name) and
+            node.func.value.id == 'unittest' and
+            node.func.attr in FUNCTION_MAPPING and
+            not has_starargs(node) and
+            not node.keywords and
+            len(node.args) == 1
     ):
         func = functools.partial(
             replace_name,

diff --git a/tests/features/unittest_aliases_test.py b/tests/features/unittest_aliases_test.py
@@ -7,7 +7,7 @@
 
 
 @pytest.mark.parametrize(
-    ('s',),
+    's',
     (
         pytest.param(
             'class ExampleTests:\n'
@@ -16,8 +16,8 @@
             id='not a deprecated alias',
         ),
         # https://github.com/asottile/pyupgrade/issues/749#
-        ('unittest.makeSuite(Tests, "arg")',),
-        ('unittest.makeSuite(Tests, prefix="arg")',),
+        ('unittest.makeSuite(Tests, "arg")'),
+        ('unittest.makeSuite(Tests, prefix="arg")'),
     ),
 )
 def test_fix_unittest_aliases_noop(s):