TST: Add a unit test.
TST: Tweak the first test case, and add another.