修改主框架，增加单元测试框架
change the main.py file and the tests.