def test_guess_categories_fails_without_a_magpie_url(g_m_u): g_m_u.return_value = None obj = StubObj({}, {}) eng = DummyEng() assert guess_categories(obj, eng) is None assert obj.extra_data == {}
def test_guess_categories_fails_without_a_magpie_url(g_m_u): g_m_u.return_value = None obj = MockObj({}, {}) eng = MockEng() assert guess_categories(obj, eng) is None assert obj.extra_data == {}
def test_guess_categories_accepts_over_point_25(j_a_r, g_m_u): j_a_r.return_value = { 'labels': [ ('foo', 0.25), ], } g_m_u.return_value = 'https://magpie.inspirehep.net/predict' obj = MockObj({}, {}) eng = MockEng() assert guess_categories(obj, eng) is None assert obj.extra_data['categories_prediction'] == { 'categories': [ { 'label': 'foo', 'score': 0.25, 'accept': True, }, ], }
def test_guess_categories_accepts_over_point_25(j_a_r, g_m_u): j_a_r.return_value = { 'labels': [ ('foo', 0.25), ], } g_m_u.return_value = 'https://magpie.inspirehep.net/predict' obj = StubObj({}, {}) eng = DummyEng() assert guess_categories(obj, eng) is None assert obj.extra_data['categories_prediction'] == { 'categories': [ { 'label': 'foo', 'score': 0.25, 'accept': True, }, ], }
def test_guess_categories_filters_under_point_22(j_a_r, g_m_u): j_a_r.return_value = { 'labels': [ ('foo', 0.21), ('bar', 0.22), ], } g_m_u.return_value = 'https://magpie.inspirehep.net/predict' obj = MockObj({}, {}) eng = MockEng() assert guess_categories(obj, eng) is None assert obj.extra_data['categories_prediction'] == { 'categories': [ { 'label': 'bar', 'score': 0.22, 'accept': False, }, ], }
def test_guess_categories_filters_under_point_22(j_a_r, g_m_u): j_a_r.return_value = { 'labels': [ ('foo', 0.21), ('bar', 0.22), ], } g_m_u.return_value = 'https://magpie.inspirehep.net/predict' obj = StubObj({}, {}) eng = DummyEng() assert guess_categories(obj, eng) is None assert obj.extra_data['categories_prediction'] == { 'categories': [ { 'label': 'bar', 'score': 0.22, 'accept': False, }, ], }