Exemplo n.º 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,
    )
Exemplo n.º 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))
Exemplo n.º 3
0
 def test_format_code_with_unicode(self):
     self.assertEqual("x = 'abc' \\\n    'ö'\n",
                      pyformat.format_code('x = "abc" \\\n"ö"\n'))
Exemplo n.º 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))
Exemplo n.º 5
0
 def test_format_code_without_aggressive(self):
     self.assertEqual(
         'import os\nTrue\n',
         pyformat.format_code('import os\nTrue\n', aggressive=False))
Exemplo n.º 6
0
 def test_format_code(self):
     self.assertEqual("x = 'abc' \\\n    'next'\n",
                      pyformat.format_code('x = "abc" \\\n"next"\n'))
Exemplo n.º 7
0
 def test_format_code_with_unicode(self):
     self.assertEqual("x = 'abc' \\\n    'ö'\n",
                      pyformat.format_code(
                          'x = "abc" \\\n"ö"\n'))
Exemplo n.º 8
0
 def test_format_code_without_aggressive(self):
     self.assertEqual('import os\nTrue\n',
                      pyformat.format_code(
                          'import os\nTrue\n',
                          aggressive=False))
Exemplo n.º 9
0
 def test_format_code(self):
     self.assertEqual("x = 'abc' \\\n    'next'\n",
                      pyformat.format_code(
                          'x = "abc" \\\n"next"\n'))