nit: I would just do `class DAGSchedulerSuite extends ...`, and spill over to new lines normally