Fix formatting for 621

ilyannn · Mar 29, 2022 · cfd9a65 · cfd9a65
1 parent 49136bd
commit cfd9a65
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/lc_0621_task_scheduler.py b/lc_0621_task_scheduler.py
@@ -29,10 +29,7 @@ def setUp(self) -> None:
     def test_examples(self):
         assert self.f(["A", "A", "A", "B", "B", "B"], 2) == 8
         assert self.f(["A", "A", "A", "B", "B", "B"], 0) == 6
-        assert (
-            self.f(["A", "A", "A", "A", "A", "A", "B", "C", "D", "E", "F", "G"], 2)
-            == 16
-        )
+        assert self.f(["A"] * 6 + ["B", "C", "D", "E", "F", "G"], 2) == 16
 
     def test_simple(self):
         assert self.f(["A"], 100) == 1