Add some test for corner cases.