Move all test functions into a test class