Fixed the unit test for a more general case.  Refs #6862.