Skip to content

Commit 66edbf7

Browse files
Dsantra92Dsantra92
authored
Patch/fix ci (#154)
* Fix end errors * remove ml-1m from CI * Remove all MovieLens tests from CI
1 parent 28c6840 commit 66edbf7

File tree

2 files changed

+125
-123
lines changed

2 files changed

+125
-123
lines changed

‎test/datasets/graphs.jl

Lines changed: 0 additions & 122 deletions
Original file line numberDiff line numberDiff line change
@@ -173,125 +173,3 @@ Sys.iswindows() || @testset "OGBn-mag" begin
173173
end
174174
end
175175

176-
@testset "ml-100k" begin
177-
data = MovieLens("100k")
178-
@test length(data) == 1
179-
180-
g = data[1]
181-
@test g == data[:]
182-
@test g isa MLDatasets.HeteroGraph
183-
184-
num_nodes = Dict(
185-
"movie" => 1682,
186-
"user" => 943,
187-
)
188-
num_edges = Dict(
189-
("user", "rating", "movie") => 200000
190-
)
191-
192-
for type in keys(num_nodes)
193-
@test type ∈ g.node_types
194-
@test g.num_nodes[type] == num_nodes[type]
195-
node_data = get(g.node_data, type, nothing)
196-
@test !isnothing(node_data)
197-
for (key, val) in node_data
198-
@test key ∈ [:release_date, :genres, :age, :occupation, :zipcode, :gender]
199-
@test size(val)[end] == num_nodes[type]
200-
end
201-
end
202-
203-
for type in keys(num_edges)
204-
@test type ∈ g.edge_types
205-
@test g.num_edges[type] == num_edges[type]
206-
@test length(g.edge_indices[type][1]) == num_edges[type]
207-
@test length(g.edge_indices[type][2]) == num_edges[type]
208-
edge_data = g.edge_data[type]
209-
for (key, val) in edge_data
210-
@test key in [:timestamp, :rating]
211-
@test ndims(val) == 1
212-
@test size(val)[end] == num_edges[type]
213-
end
214-
end
215-
end
216-
217-
@testset "ml-1m" begin
218-
data = MovieLens("1m")
219-
@test length(data) == 1
220-
221-
g = data[1]
222-
@test g == data[:]
223-
@test g isa MLDatasets.HeteroGraph
224-
225-
num_nodes = Dict(
226-
"movie" => 3883,
227-
"user" => 6040
228-
)
229-
num_edges = Dict(
230-
("user", "rating", "movie") => 2000418
231-
)
232-
233-
for type in keys(num_nodes)
234-
@test type ∈ g.node_types
235-
@test g.num_nodes[type] == num_nodes[type]
236-
node_data = get(g.node_data, type, nothing)
237-
@test !isnothing(node_data)
238-
for (key, val) in node_data
239-
@test key ∈ [:genres, :age, :occupation, :zipcode, :gender]
240-
@test size(val)[end] == num_nodes[type]
241-
end
242-
end
243-
244-
for type in keys(num_edges)
245-
@test type ∈ g.edge_types
246-
@test g.num_edges[type] == num_edges[type]
247-
@test length(g.edge_indices[type][1]) == num_edges[type]
248-
@test length(g.edge_indices[type][2]) == num_edges[type]
249-
edge_data = g.edge_data[type]
250-
for (key, val) in edge_data
251-
@test key in [:timestamp, :rating]
252-
@test ndims(val) == 1
253-
@test size(val)[end] == num_edges[type]
254-
end
255-
end
256-
end
257-
258-
@testset "ml-latest-small" begin
259-
data = MovieLens("latest-small")
260-
@test length(data) == 1
261-
262-
g = data[1]
263-
@test g == data[:]
264-
@test g isa MLDatasets.HeteroGraph
265-
266-
num_nodes = Dict(
267-
"tag" => 3683,
268-
"movie" => 9742,
269-
"user" => 610
270-
)
271-
num_edges = Dict(
272-
("user", "rating", "movie") => 201672,
273-
("user", "tag", "movie") => 7366
274-
)
275-
276-
for type in keys(num_nodes)
277-
@test type ∈ g.node_types
278-
@test g.num_nodes[type] == num_nodes[type]
279-
node_data = get(g.node_data, type, nothing)
280-
isnothing(node_data) || for (key, val) in node_data
281-
@test size(val)[end] == num_nodes[type]
282-
end
283-
end
284-
285-
for type in keys(num_edges)
286-
@test type ∈ g.edge_types
287-
@test g.num_edges[type] == num_edges[type]
288-
@test length(g.edge_indices[type][1]) == num_edges[type]
289-
@test length(g.edge_indices[type][2]) == num_edges[type]
290-
edge_data = g.edge_data[type]
291-
for (key, val) in edge_data
292-
@test key in [:timestamp, :tag_name, :rating]
293-
@test ndims(val) == 1
294-
@test size(val)[end] == num_edges[type]
295-
end
296-
end
297-
end

‎test/datasets/graphs_no_ci.jl

Lines changed: 125 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,129 @@ end
112112
@test sum(count.([d.graph_data.train_mask, d.graph_data.test_mask, d.graph_data.val_mask])) == length(d)
113113
end
114114

115+
@testset "ml-100k" begin
116+
data = MovieLens("100k")
117+
@test length(data) == 1
118+
119+
g = data[1]
120+
@test g == data[:]
121+
@test g isa MLDatasets.HeteroGraph
122+
123+
num_nodes = Dict(
124+
"movie" => 1682,
125+
"user" => 943,
126+
)
127+
num_edges = Dict(
128+
("user", "rating", "movie") => 200000
129+
)
130+
131+
for type in keys(num_nodes)
132+
@test type ∈ g.node_types
133+
@test g.num_nodes[type] == num_nodes[type]
134+
node_data = get(g.node_data, type, nothing)
135+
@test !isnothing(node_data)
136+
for (key, val) in node_data
137+
@test key ∈ [:release_date, :genres, :age, :occupation, :zipcode, :gender]
138+
@test size(val)[end] == num_nodes[type]
139+
end
140+
end
141+
142+
for type in keys(num_edges)
143+
@test type ∈ g.edge_types
144+
@test g.num_edges[type] == num_edges[type]
145+
@test length(g.edge_indices[type][1]) == num_edges[type]
146+
@test length(g.edge_indices[type][2]) == num_edges[type]
147+
edge_data = g.edge_data[type]
148+
for (key, val) in edge_data
149+
@test key in [:timestamp, :rating]
150+
@test ndims(val) == 1
151+
@test size(val)[end] == num_edges[type]
152+
end
153+
end
154+
end
155+
156+
@testset "ml-latest-small" begin
157+
data = MovieLens("latest-small")
158+
@test length(data) == 1
159+
160+
g = data[1]
161+
@test g == data[:]
162+
@test g isa MLDatasets.HeteroGraph
163+
164+
num_nodes = Dict(
165+
"tag" => 3683,
166+
"movie" => 9742,
167+
"user" => 610
168+
)
169+
num_edges = Dict(
170+
("user", "rating", "movie") => 201672,
171+
("user", "tag", "movie") => 7366
172+
)
173+
174+
for type in keys(num_nodes)
175+
@test type ∈ g.node_types
176+
@test g.num_nodes[type] == num_nodes[type]
177+
node_data = get(g.node_data, type, nothing)
178+
isnothing(node_data) || for (key, val) in node_data
179+
@test size(val)[end] == num_nodes[type]
180+
end
181+
end
182+
183+
for type in keys(num_edges)
184+
@test type ∈ g.edge_types
185+
@test g.num_edges[type] == num_edges[type]
186+
@test length(g.edge_indices[type][1]) == num_edges[type]
187+
@test length(g.edge_indices[type][2]) == num_edges[type]
188+
edge_data = g.edge_data[type]
189+
for (key, val) in edge_data
190+
@test key in [:timestamp, :tag_name, :rating]
191+
@test ndims(val) == 1
192+
@test size(val)[end] == num_edges[type]
193+
end
194+
end
195+
end
196+
197+
@testset "ml-1m" begin
198+
data = MovieLens("1m")
199+
@test length(data) == 1
200+
201+
g = data[1]
202+
@test g == data[:]
203+
@test g isa MLDatasets.HeteroGraph
204+
205+
num_nodes = Dict(
206+
"movie" => 3883,
207+
"user" => 6040
208+
)
209+
num_edges = Dict(
210+
("user", "rating", "movie") => 2000418
211+
)
212+
213+
for type in keys(num_nodes)
214+
@test type ∈ g.node_types
215+
@test g.num_nodes[type] == num_nodes[type]
216+
node_data = get(g.node_data, type, nothing)
217+
@test !isnothing(node_data)
218+
for (key, val) in node_data
219+
@test key ∈ [:genres, :age, :occupation, :zipcode, :gender]
220+
@test size(val)[end] == num_nodes[type]
221+
end
222+
end
223+
224+
for type in keys(num_edges)
225+
@test type ∈ g.edge_types
226+
@test g.num_edges[type] == num_edges[type]
227+
@test length(g.edge_indices[type][1]) == num_edges[type]
228+
@test length(g.edge_indices[type][2]) == num_edges[type]
229+
edge_data = g.edge_data[type]
230+
for (key, val) in edge_data
231+
@test key in [:timestamp, :rating]
232+
@test ndims(val) == 1
233+
@test size(val)[end] == num_edges[type]
234+
end
235+
end
236+
end
237+
115238
@testset "ml-10m" begin
116239
data = MovieLens("10m")
117240
@test length(data) == 1
@@ -135,7 +258,8 @@ end
135258
@test g.num_nodes[type] == num_nodes[type]
136259
node_data = get(g.node_data, type, nothing)
137260
isnothing(node_data) || for (key, val) in node_data
138-
@test size(val)[end] == num_nodes[type]
261+
@test size(val)[end] == num_nodes[type]
262+
end
139263
end
140264

141265
for type in keys(num_edges)

0 commit comments

Comments
 (0)