@@ -823,7 +823,7 @@ def another_function():
823823 return 42"""
824824
825825 # Test with None formatter
826- result = format_generated_code (test_code , None )
826+ result = format_generated_code (test_code , [ "disabled" ] )
827827 # Multiple newlines (3+) are reduced to 2
828828 expected = """import os
829829
@@ -859,7 +859,7 @@ def test_format_generated_code_disabled_case_insensitive():
859859
860860def test_format_generated_code_empty_string ():
861861 """Test format_generated_code with empty string."""
862- result = format_generated_code ("" , None )
862+ result = format_generated_code ("" , [ "disabled" ] )
863863 assert result == ""
864864
865865 result = format_generated_code ("" , ["disabled" ])
@@ -1340,7 +1340,7 @@ def func1():
13401340def func2():
13411341 pass"""
13421342
1343- result = format_generated_code (test_code , None )
1343+ result = format_generated_code (test_code , [ "disabled" ] )
13441344 # Should have at most two consecutive newlines
13451345 assert "\n \n \n " not in result
13461346 assert "import os\n \n " in result
@@ -1395,7 +1395,7 @@ def test_format_generated_code_unicode():
13951395 message = "Hello, 世界! 🌍"
13961396 return message"""
13971397
1398- result = format_generated_code (test_code , None )
1398+ result = format_generated_code (test_code , [ "disabled" ] )
13991399 assert "Hello, 世界! 🌍" in result
14001400
14011401
0 commit comments