Add more tests for custom attribute methods