add test file

the test file is the main function