Improve unit test.
Further improve already awesome unit test.