Function test_set_best_config

tests/test_experiments.py:8–16

()

import os

def test_set_best_config():
    args = get_default_args(task="node_classification", dataset="citeseer", model="gat")
    args.model = args.model[0]
    args.dataset = args.dataset[0]
    args = set_best_config(args)

    assert args.lr == 0.005
    assert args.epochs == 1000
    assert args.weight_decay == 0.001


def test_train():