Add unit tests for refactored scheduler initialization.