Function test_experiment

tests/test_experiments.py:36–40

()

def test_experiment():
    results = experiment(task="node_classification", dataset="cora", model="gcn", hidden_size=32, epochs=10, cpu=True)

    assert ("cora", "gcn") in results
    assert results[("cora", "gcn")][0]["test_acc"] > 0


def test_auto_experiment():