1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
import pytest
def test_ba_shapes(get_dataset):
with pytest.warns(UserWarning, match="is deprecated"):
dataset = get_dataset(name='BAShapes')
assert str(dataset) == 'BAShapes()'
assert len(dataset) == 1
assert dataset.num_features == 10
assert dataset.num_classes == 4
data = dataset[0]
assert len(data) == 5
assert data.edge_index.size(1) >= 1120
assert data.x.size() == (700, 10)
assert data.y.size() == (700, )
assert data.expl_mask.sum() == 60
assert data.edge_label.sum() == 960
|
