テストコードを修正

関数に抽出