Update assert functions to remove duplicated code via assert_op in test_helper.py