Use np.testing.assert_almost_equal instead of map