add inference tests