Add type mapping test