assert puzzle and ability to pass single argument to assert