Merge pull request #214 from JuliaSymbolics/myb/num

Add skips in number_methods

Author: YingboMa

Project.toml

@@ -1,7 +1,7 @@
 name = "SymbolicUtils"
 uuid = "d1185830-fcd6-423d-90d6-eec64667417b"
 authors = ["Shashi Gowda"]
-version = "0.8.3"
+version = "0.8.4"
 
 [deps]
 AbstractAlgebra = "c3fe647b-3220-5bb0-a1ea-a7954cac585d"

src/methods.jl

@@ -34,14 +34,16 @@ islike(a, T) = symtype(a) <: T
 function number_methods(T, rhs1, rhs2, options=nothing)
     exprs = []
 
-    skip_basics = !isnothing(options) ? options == :skipbasics : false
+    skip_basics = options !== nothing ? options == :skipbasics : false
+    skips = Meta.isexpr(options, [:vcat, :hcat, :vect]) ? Set(options.args) : []
     basic_monadic = [-, +]
     basic_diadic = [+, -, *, /, \, ^]
 
     rhs2 = :($assert_like(f, Number, a, b); $rhs2)
     rhs1 = :($assert_like(f, Number, a); $rhs1)
 
     for f in (skip_basics ? diadic : vcat(basic_diadic, diadic))
+        nameof(f) in skips && continue
         for S in previously_declared_for
             push!(exprs, quote
                       (f::$(typeof(f)))(a::$T, b::$S) = $rhs2
@@ -62,6 +64,7 @@ function number_methods(T, rhs1, rhs2, options=nothing)
     end
 
     for f in (skip_basics ? monadic : vcat(basic_monadic, monadic))
+        nameof(f) in skips && continue
         push!(exprs, :((f::$(typeof(f)))(a::$T)   = $rhs1))
     end
     push!(exprs, :(push!($previously_declared_for, $T)))