modify tests.py
modify test structure
Merge pull request #7 from luocheng/master

justify tests structure