Add a unit test for custom Map sub-class