Add more testing

[ltitanb]

We should add unit and integration testing in the repo, and make sure we have a good variety of test cases. The PBS module has already some initial testing in /tests but they need to be tidied up and expanded