get_rules -> getindex

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

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/ExprRules.jl b/src/ExprRules.jl
@@ -12,7 +12,6 @@ export
 
         @ruleset,
         @digits,
-        get_rules,
         max_arity,
         depth,
         isterminal,
@@ -112,8 +111,9 @@ function _parse_rule!(v::Vector{Any}, ex::Expr)
     end
 end
 
+Base.getindex(ruleset::RuleSet, typ::Symbol) = ruleset.bytype[typ]
+
 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,8 +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 ruleset[:R] == ruleset.bytype[:R]
+    @test ruleset[:I] == ruleset.bytype[:I]
     @test contains_returntype(rulenode, ruleset, :R)
     @test contains_returntype(rulenode, ruleset, :I)
     @test !contains_returntype(rulenode, ruleset, :B)