update test.py add some tests