Skip to content
Permalink
Browse files

test coverage

  • Loading branch information...
rusty1s committed Mar 3, 2019
1 parent 8deb289 commit 673a946f4911b497c0bd6a39b8fa93804e9e69f9
Showing with 6 additions and 1 deletion.
  1. +1 −0 .coveragerc
  2. +5 −1 test/datasets/test_enzymes.py
@@ -4,6 +4,7 @@ omit=
torch_geometric/datasets/*
torch_geometric/read/*
torch_geometric/data/extract.py
torch_geometric/nn/data_parallel.py
[report]
exclude_lines =
pragma: no cover
@@ -8,7 +8,7 @@
import pytest
import torch
from torch_geometric.datasets import TUDataset
from torch_geometric.data import DataLoader, DenseDataLoader
from torch_geometric.data import DataLoader, DenseDataLoader, DataListLoader
from torch_geometric.transforms import ToDense


@@ -49,6 +49,10 @@ def test_enzymes():
assert not data.contains_self_loops()
assert data.is_undirected()

loader = DataListLoader(dataset, batch_size=len(dataset))
for data_list in loader:
assert len(data_list) == 600

dataset.transform = ToDense(num_nodes=126)
loader = DenseDataLoader(dataset, batch_size=len(dataset))
for data in loader:

0 comments on commit 673a946

Please sign in to comment.
You can’t perform that action at this time.