From 583cad518616d08c38fd0c0d8ebf61839a518078 Mon Sep 17 00:00:00 2001 From: Hatem Helal Date: Thu, 15 Sep 2022 13:41:18 +0100 Subject: [PATCH] rtest for repr --- test/data/test_summary.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/data/test_summary.py b/test/data/test_summary.py index c35dcf118a7d6..cfa675eff07ef 100644 --- a/test/data/test_summary.py +++ b/test/data/test_summary.py @@ -27,6 +27,7 @@ def test_summary(): seed_everything(0) dataset = FakeDataset(num_graphs=10) summary = dataset.summary() + assert "FakeDataset" in str(summary) assert summary.num_graphs == 10 num_nodes = torch.tensor([d.num_nodes for d in dataset]).float() num_edges = torch.tensor([d.num_edges for d in dataset]).float() @@ -50,6 +51,7 @@ def test_hetero_summary(): seed_everything(0) dataset = FakeHeteroDataset(num_graphs=10) summary = dataset.summary() + assert "FakeHeteroDataset" in str(summary) assert summary.num_graphs == 10 num_nodes = torch.tensor([d.num_nodes for d in dataset]).float() num_edges = torch.tensor([d.num_edges for d in dataset]).float()