Added tests for the new functionality.


2 files changed