Add test dir and test.py