Added additional test and tweaked assert method signiture.