Fix travis build failure + use assertEqual