get_rules

sisl · Aug 18, 2017 · 0f4bd91 · 0f4bd91
1 parent db6a9c0
commit 0f4bd91
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/ExprRules.jl b/src/ExprRules.jl
@@ -12,6 +12,7 @@ export
 
         @ruleset,
         @digits,
+        get_rules,
         max_arity,
         depth,
         isterminal,
@@ -112,6 +113,7 @@ function _parse_rule!(v::Vector{Any}, ex::Expr)
 end
 
 nonterminals(ruleset::RuleSet) = collect(keys(ruleset.bytype))
+get_rules(ruleset::RuleSet, typ::Symbol) = ruleset.bytype[typ]
 return_type(ruleset::RuleSet, rule_index::Int) = ruleset.types[rule_index]
 child_types(ruleset::RuleSet, rule_index::Int) = ruleset.childtypes[rule_index]
 isterminal(ruleset::RuleSet, rule_index::Int) = ruleset.isterminal[rule_index]

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -24,6 +24,8 @@ let
 
     rulenode = RuleNode(1, [RuleNode(2)])
     get_executable(rulenode, ruleset)
+    @test get_rules(ruleset, :R) == ruleset.bytype[:R]
+    @test get_rules(ruleset, :I) == ruleset.bytype[:I]
     @test contains_returntype(rulenode, ruleset, :R)
     @test contains_returntype(rulenode, ruleset, :I)
     @test !contains_returntype(rulenode, ruleset, :B)
@@ -130,5 +132,7 @@ let
 
     srand(4)
     rulenode = RuleNode(3, [RuleNode(4), RuleNode(5)])
-    loc = sample(NodeLoc, rulenode, :Real, ruleset)
+    for i in 1 : 10
+        loc = sample(NodeLoc, rulenode, :Real, ruleset)
+    end
 end