Add simple test on map function