add assert in test