fix source() == source() bug; mv children() out to tests

Author: ablaom

src/composition/learning_networks/inspection.jl

@@ -114,17 +114,6 @@ train_args(::Source) = []
 train_args(N::Node{<:Machine}) = N.machine.args
 train_args(N::Node{Nothing}) = []
 
-"""
-    children(N::AbstractNode, y::AbstractNode)
-
-List all (immediate) children of node `N` in the ancestor graph of `y`
-(training edges included).
-
-"""
-children(N::AbstractNode, y::AbstractNode) = filter(nodes(y)) do W
-    N in args(W) || N in train_args(W)
-end |> unique
-
 """
     lower_bound(type_itr)
 

src/sources.jl

@@ -30,6 +30,9 @@ mutable struct Source <: AbstractNode
     scitype::DataType
 end
 
+# To ensure `source() != source()`:
+MMI.is_same_except(s1::Source, s2::Source; kwargs...) = s1 === s2
+
 """
     Xs = source(X=nothing)
 

test/composition/learning_networks/inspection.jl

@@ -3,8 +3,20 @@ module TestLearningCompositesInspection
 using Test
 using MLJBase
 using ..Models
+import MLJModelInterface as MMI
 
-KNNRegressor()
+"""
+    children(N::AbstractNode, y::AbstractNode)
+
+List all (immediate) children of node `N` in the ancestor graph of `y`
+(training edges included).
+
+"""
+children(N::AbstractNode, y::AbstractNode) = filter(nodes(y)) do Z
+    t = MMI.isrepresented(N, MLJBase.args(Z)) ||
+        MMI.isrepresented(N, MLJBase.train_args(Z))
+    @show Z t
+end |> unique
 
 @constant X = source()
 @constant y = source()
@@ -36,12 +48,12 @@ knnM = machine(knn, W, y)
 @test Set(machines(yhat)) == Set([knnM, hotM])
 @test Set(MLJBase.args(yhat)) == Set([W, ])
 @test Set(MLJBase.train_args(yhat)) == Set([W, y])
-@test Set(MLJBase.children(X, all)) == Set([W, K])
+@test Set(children(X, all)) == Set([W, K])
 
 @constant Q = 2X
 @constant R = 3X
 @constant S = glb(X, Q, R)
-@test Set(MLJBase.children(X, S)) == Set([Q, R, S])
+@test Set(children(X, S)) == Set([Q, R, S])
 @test MLJBase.lower_bound([Int, Float64]) == Union{}
 @test MLJBase.lower_bound([Int, Integer]) == Int
 @test MLJBase.lower_bound([Int, Integer]) == Int

test/sources.jl

@@ -13,6 +13,7 @@ Xs = source(X)
 rebind!(Xs, nothing)
 @test isempty(Xs)
 @test Xs.scitype == Nothing
+@test source() != source()
 
 end
 true