Add tests for enums