Adds a test for the new feature