add test for map