Improved tests with a test suite