Refactor answer model test