Ejemplos de guess_experiments en Python

Ejemplo n.º 1

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

Ejemplo n.º 2

Archivo: test_workflows_tasks_magpie.py Proyecto: bittirousku/inspire-next

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

Ejemplo n.º 3

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

Ejemplo n.º 4

Archivo: test_workflows_tasks_magpie.py Proyecto: david-caro/inspire-next

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