better python test mapping