Add some assert in test