I added a generic unit test that covers all cases. Let me know if you think something else is more appropriate.