Add tests for policy classes