Improved test, separate assertions to each test.