Support running pytest directly

This MR ensures pytest can be run directly using pytest instead of python -m pytest. This is useful for integrating the tests in editors like vscode.

The requirement for running pytest directly is that the modules are in the python path. This is not an issue when running python -m pytest, since python adds the current working directory to the path automatically