protected function onNotSuccessfulTest(\Exception $e)
 {
     if ($e instanceof PredictionException) {
         $e = new \PHPUnit_Framework_AssertionFailedError($e->getMessage(), $e->getCode(), $e);
     }
     return parent::onNotSuccessfulTest($e);
 }