clean up the tests and add more data assertions