Skip to content

Commit c482ec9

Browse files
Merge pull request #132 from JuliaML/cl/fix
KarateClub <: AbstractDataset
2 parents d5f5178 + d74e6f4 commit c482ec9

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

src/datasets/graphs/karateclub.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ One node per unique label is used as training data.
2525
2626
[4]: [NetworkX Zachary's Karate Club Dataset](https://networkx.org/documentation/stable/_modules/networkx/generators/social.html#karate_club_graph)
2727
"""
28-
struct KarateClub
28+
struct KarateClub <: AbstractDataset
2929
metadata::Dict{String, Any}
3030
graphs::Vector{Graph}
3131
end

test/datasets/graphs.jl

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11

22
@testset "Cora" begin
33
data = Cora()
4+
@test data isa AbstractDataset
45
@test length(data) == 1
56
g = data[1]
67
@test g === data[:]
@@ -25,6 +26,7 @@ end
2526

2627
@testset "CiteSeer" begin
2728
data = CiteSeer()
29+
@test data isa AbstractDataset
2830
@test length(data) == 1
2931
g = data[1]
3032
@test g === data[:]
@@ -49,6 +51,7 @@ end
4951

5052
@testset "PubMed" begin
5153
data = PubMed()
54+
@test data isa AbstractDataset
5255
@test length(data) == 1
5356
g = data[1]
5457
@test g === data[:]
@@ -73,6 +76,7 @@ end
7376

7477
@testset "KarateClub" begin
7578
data = KarateClub()
79+
@test data isa AbstractDataset
7680
@test length(data) == 1
7781
g = data[1]
7882
@test g === data[:]
@@ -98,6 +102,7 @@ end
98102

99103
@testset "PolBlogs" begin
100104
data = PolBlogs()
105+
@test data isa AbstractDataset
101106
@test length(data) == 1
102107
g = data[1]
103108
@test g === data[:]

0 commit comments

Comments
 (0)