add test data and python code