def test_guess_experiments_fails_without_a_magpie_url(g_m_u): g_m_u.return_value = None obj = MockObj({}, {}) eng = MockEng() assert guess_experiments(obj, eng) is None assert obj.extra_data == {}
def test_guess_experiments_fails_without_a_magpie_url(g_m_u): g_m_u.return_value = None obj = StubObj({}, {}) eng = DummyEng() assert guess_experiments(obj, eng) is None assert obj.extra_data == {}
def test_guess_experiments_filters_under_point_50(j_a_r, g_m_u): j_a_r.return_value = { 'labels': [ ('foo', 0.49), ('bar', 0.50), ], } g_m_u.return_value = 'https://magpie.inspirehep.net/predict' obj = MockObj({}, {}) eng = MockEng() assert guess_experiments(obj, eng) is None assert obj.extra_data['experiments_prediction'] == { 'experiments': [{ 'label': 'bar', 'score': 0.50, }], }
def test_guess_experiments_filters_under_point_50(j_a_r, g_m_u): j_a_r.return_value = { 'labels': [ ('foo', 0.49), ('bar', 0.50), ], } g_m_u.return_value = 'https://magpie.inspirehep.net/predict' obj = MockObj({}, {}) eng = MockEng() assert guess_experiments(obj, eng) is None assert obj.extra_data['experiments_prediction'] == { 'experiments': [ { 'label': 'bar', 'score': 0.50, } ], }