Modify test timeout

use TestNG static imports instead of using Assert class directly.