split test code to test classes