tests/composition.py

   1 class C:
   2
   3     def test(self) -> None:
   4         with patch("black.out", print):
   5             self.assertEqual(
   6                 unstyle(str(report)), "1 file reformatted, 1 file failed to reformat."
   7             )
   8             self.assertEqual(
   9                 unstyle(str(report)),
  10                 "1 file reformatted, 1 file left unchanged, 1 file failed to reformat.",
  11             )
  12             self.assertEqual(
  13                 unstyle(str(report)),
  14                 "2 files reformatted, 1 file left unchanged, "
  15                 "1 file failed to reformat.",
  16             )
  17             self.assertEqual(
  18                 unstyle(str(report)),
  19                 "2 files reformatted, 2 files left unchanged, "
  20                 "2 files failed to reformat.",
  21             )
  22             for i in (a,):
  23                 if (
  24                     # Rule 1
  25                     i % 2 == 0
  26                     # Rule 2
  27                     and i % 3 == 0
  28                 ):
  29                     while (
  30                         # Just a comment
  31                         call()
  32                         # Another
  33                     ):
  34                         print(i)
  35
  36     def omitting_trailers() -> None:
  37         get_collection(
  38             hey_this_is_a_very_long_call, it_has_funny_attributes, really=True
  39         )[OneLevelIndex]
  40         get_collection(
  41             hey_this_is_a_very_long_call, it_has_funny_attributes, really=True
  42         )[OneLevelIndex][TwoLevelIndex][ThreeLevelIndex][FourLevelIndex]
  43         d[0][1][2][3][4][5][6][7][8][9][10][11][12][13][14][15][16][17][18][19][20][21][
  44             22
  45         ]