python script for generating tests