Refactor tests

Details:
- Heavily refactor the test suite and change the directory structure