()
| 17 | |
| 18 | |
| 19 | def test_train(): |
| 20 | args = get_default_args(dataset="cora", model="gcn", epochs=10, cpu=True) |
| 21 | args.dataset = args.dataset[0] |
| 22 | args.model = args.model[0] |
| 23 | args.seed = args.seed[0] |
| 24 | result = train(args) |
| 25 | |
| 26 | assert "test_acc" in result |
| 27 | assert result["test_acc"] > 0 |
| 28 | |
| 29 | |
| 30 | def test_gen_variants(): |
no test coverage detected