Moved test files to pytest dir