/** * @param int $count * @return Test */ private function getTestWithRightAnswerCount(int $count) { $test = new Test(); for ($i = 0; $i < $count; $i++) { $var1 = new Variant(); $var1->setId($i + 1); $var1->setPrOtvet(true); $test->addVariant($var1); } return $test; }