Пример #1
0
def test_children_fetch_trials(create_db_instance):
    """Test that experiment fetch trials from children properly (adapters are muted)"""
    experiment_name = 'supernaedo2'
    root_name = 'supernaedo2'
    leaf_names = ['supernaedo2.3']

    experiment = ExperimentView(experiment_name)
    exp_node = build_trimmed_tree(experiment, root_name, leaf_names)

    assert exp_node.item.name == experiment_name
    assert exp_node.children[0].item.name == leaf_names[0]
    assert len(exp_node.children) == 1
    # 2
    # |
    # 2.3
    assert len(list(exp_node.root)) == 2

    experiment.connect_to_version_control_tree(exp_node)

    for node in exp_node.root:
        node.item._experiment.refers['adapter'] = Adapter.build([])

    query = {'status': 'completed'}
    assert len(experiment.fetch_trials(query)) == 6
    assert len(experiment._experiment._node.children[0].item.fetch_trials(query)) == 4
    assert len(experiment.fetch_trials_tree(query)) == 10
Пример #2
0
def test_init(init_full_x, create_db_instance):
    """Test if original experiment contains trial 0"""
    experiment = ExperimentView('full_x')
    pairs = get_name_value_pairs(experiment.fetch_trials({}))
    assert pairs == ((('/x', 0), ), )