Replaced assert_ by assertEqual and assertTrue