btag weight testing code