ActivityScenarioのactivityにアクセスする

getActivityのような関数が見当たりませんでしたが、以下のコードで直接Activityにアクセスできます。
Test Labでスクショを撮影したい場合にactivityへアクセスする必要があったので調べました。

        val activityScenario = ActivityScenario.launch(MainActivity::class.java)
        activityScenario.onActivity {
            ScreenShotter.takeScreenshot("top1", it)
        }