Python guess_categories Examples

Example #1

File: test_workflows_tasks_magpie.py Project: bittirousku/inspire-next

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 == {}

Example #2

File: test_workflows_tasks_magpie.py Project: rikirenz/inspire-next

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 == {}

Example #3

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,
            },
        ],
    }

Example #4

File: test_workflows_tasks_magpie.py Project: bittirousku/inspire-next

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,
            },
        ],
    }

Example #5

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,
            },
        ],
    }

Example #6

File: test_workflows_tasks_magpie.py Project: bittirousku/inspire-next

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,
            },
        ],
    }