def test_guess_keywords_fails_without_a_magpie_url(g_m_u): g_m_u.return_value = None obj = MockObj({}, {}) eng = MockEng() assert guess_keywords(obj, eng) is None assert obj.extra_data == {}
def test_guess_keywords_fails_without_a_magpie_url(g_m_u): g_m_u.return_value = None obj = StubObj({}, {}) eng = DummyEng() assert guess_keywords(obj, eng) is None assert obj.extra_data == {}
def test_guess_keywords_does_not_fail_when_request_fails(j_a_r, g_m_u): j_a_r.side_effect = requests.exceptions.RequestException() g_m_u.return_value = 'https://magpie.inspirehep.net/predict' obj = MockObj({}, {}) eng = MockEng() assert guess_keywords(obj, eng) is None assert obj.extra_data == {}
def test_guess_keywords_does_not_fail_when_request_fails(j_a_r, g_m_u): j_a_r.side_effect = requests.exceptions.RequestException() g_m_u.return_value = 'https://magpie.inspirehep.net/predict' obj = StubObj({}, {}) eng = DummyEng() assert guess_keywords(obj, eng) is None assert obj.extra_data == {}
def test_guess_keywords_accepts_over_point_09(j_a_r, g_m_u): j_a_r.return_value = { 'labels': [ ('foo', 0.09), ], } g_m_u.return_value = 'https://magpie.inspirehep.net/predict' obj = MockObj({}, {}) eng = MockEng() assert guess_keywords(obj, eng) is None assert obj.extra_data['keywords_prediction'] == { 'keywords': [ { 'label': 'foo', 'score': 0.09, 'accept': True, }, ] }
def test_guess_keywords_accepts_over_point_09(j_a_r, g_m_u): j_a_r.return_value = { 'labels': [ ('foo', 0.09), ], } g_m_u.return_value = 'https://magpie.inspirehep.net/predict' obj = StubObj({}, {}) eng = DummyEng() assert guess_keywords(obj, eng) is None assert obj.extra_data['keywords_prediction'] == { 'keywords': [ { 'label': 'foo', 'score': 0.09, 'accept': True, }, ] }
def test_guess_keywords_considers_only_first_ten(j_a_r, g_m_u): j_a_r.return_value = { 'labels': [ ('k01', 1.00), ('k02', 1.00), ('k03', 1.00), ('k04', 1.00), ('k05', 1.00), ('k06', 1.00), ('k07', 1.00), ('k08', 1.00), ('k09', 1.00), ('k10', 1.00), ('k11', 1.00), # Will be ignored. ], } g_m_u.return_value = 'https://magpie.inspirehep.net/predict' obj = StubObj({}, {}) eng = DummyEng() assert guess_keywords(obj, eng) is None assert obj.extra_data['keywords_prediction'] == { 'keywords': [ {'label': 'k01', 'score': 1.00, 'accept': True}, {'label': 'k02', 'score': 1.00, 'accept': True}, {'label': 'k03', 'score': 1.00, 'accept': True}, {'label': 'k04', 'score': 1.00, 'accept': True}, {'label': 'k05', 'score': 1.00, 'accept': True}, {'label': 'k06', 'score': 1.00, 'accept': True}, {'label': 'k07', 'score': 1.00, 'accept': True}, {'label': 'k08', 'score': 1.00, 'accept': True}, {'label': 'k09', 'score': 1.00, 'accept': True}, {'label': 'k10', 'score': 1.00, 'accept': True}, ], }
def test_guess_keywords_considers_only_first_ten(j_a_r, g_m_u): j_a_r.return_value = { 'labels': [ ('k01', 1.00), ('k02', 1.00), ('k03', 1.00), ('k04', 1.00), ('k05', 1.00), ('k06', 1.00), ('k07', 1.00), ('k08', 1.00), ('k09', 1.00), ('k10', 1.00), ('k11', 1.00), # Will be ignored. ], } g_m_u.return_value = 'https://magpie.inspirehep.net/predict' obj = StubObj({}, {}) eng = DummyEng() assert guess_keywords(obj, eng) is None assert obj.extra_data['keywords_prediction'] == { 'keywords': [ {'label': 'k01', 'score': 1.00, 'accept': True}, {'label': 'k02', 'score': 1.00, 'accept': True}, {'label': 'k03', 'score': 1.00, 'accept': True}, {'label': 'k04', 'score': 1.00, 'accept': True}, {'label': 'k05', 'score': 1.00, 'accept': True}, {'label': 'k06', 'score': 1.00, 'accept': True}, {'label': 'k07', 'score': 1.00, 'accept': True}, {'label': 'k08', 'score': 1.00, 'accept': True}, {'label': 'k09', 'score': 1.00, 'accept': True}, {'label': 'k10', 'score': 1.00, 'accept': True}, ], }