Updated the tests with proper assertion methods.