add the test.py file

Review request sent by eddie, diff file test