Create test.py

I have just compose a demo test file