add more test assertions