Example #1
0
def _pyformat(code: str, **params):
    import pyformat

    return pyformat.format_code(
        code,
        aggressive=True,
        apply_config=False,
        remove_all_unused_imports=False,
        remove_unused_variables=False,
    )
Example #2
0
 def test_format_code_with_remove_unused_variables(self):
     self.assertEqual(
         'def test():\n    return 42\n',
         pyformat.format_code('def test():\n    x = 4\n    return 42',
                              aggressive=True,
                              remove_unused_variables=True))
Example #3
0
 def test_format_code_with_unicode(self):
     self.assertEqual("x = 'abc' \\\n    'ö'\n",
                      pyformat.format_code('x = "abc" \\\n"ö"\n'))
Example #4
0
 def test_format_code_with_remove_all_unused_imports(self):
     self.assertEqual(
         "x = 'abc' \\\n    'ö'\n",
         pyformat.format_code('import my_module\nx = "abc" \\\n"ö"\n',
                              aggressive=True,
                              remove_all_unused_imports=True))
Example #5
0
 def test_format_code_without_aggressive(self):
     self.assertEqual(
         'import os\nTrue\n',
         pyformat.format_code('import os\nTrue\n', aggressive=False))
Example #6
0
 def test_format_code(self):
     self.assertEqual("x = 'abc' \\\n    'next'\n",
                      pyformat.format_code('x = "abc" \\\n"next"\n'))
Example #7
0
 def test_format_code_with_unicode(self):
     self.assertEqual("x = 'abc' \\\n    'ö'\n",
                      pyformat.format_code(
                          'x = "abc" \\\n"ö"\n'))
Example #8
0
 def test_format_code_without_aggressive(self):
     self.assertEqual('import os\nTrue\n',
                      pyformat.format_code(
                          'import os\nTrue\n',
                          aggressive=False))
Example #9
0
 def test_format_code(self):
     self.assertEqual("x = 'abc' \\\n    'next'\n",
                      pyformat.format_code(
                          'x = "abc" \\\n"next"\n'))