move test.py into test folder