add python test data generator