Add tests

Herb-AI · Nov 26, 2024 · e3a5874 · e3a5874
1 parent f6cea02
commit e3a5874
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 6 deletions.
diff --git a/Project.toml b/Project.toml
@@ -9,9 +9,3 @@ AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"
 [compat]
 AbstractTrees = "0.4.5"
 julia = "^1.8"
-
-[extras]
-Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
-
-[targets]
-test = ["Test"]
diff --git a/src/HerbCore.jl b/src/HerbCore.jl
@@ -32,4 +32,8 @@ export
        have_same_shape, AbstractConstraint,
        AbstractGrammar
 
+# AbstractTrees interface
+children,
+nodevalue
+
 end # module HerbCore
diff --git a/test/Project.toml b/test/Project.toml
@@ -0,0 +1,3 @@
+[deps]
+AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"
+Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1,7 +1,17 @@
+using AbstractTrees: children, nodevalue, treeheight
 using HerbCore
 using Test
 
 @testset "HerbCore.jl" verbose=true begin
+    @testset "AbstractTrees Interface" begin
+        @show typeof(nodevalue(RuleNode(1)))
+        @test nodevalue(RuleNode(1)) == 1
+        @test isempty(children(RuleNode(1)))
+        @test length(children(RuleNode(1, [RuleNode(2), RuleNode(2)]))) == 2
+        @test treeheight(RuleNode(1)) == 0
+        @test treeheight(RuleNode(1, [RuleNode(2), RuleNode(2)])) == 1
+    end
+
     @testset "RuleNode tests" begin
         @testset "Equality tests" begin
             @test RuleNode(1) == RuleNode(1)