Try to simplify numbers with integer values to integers

Author: hersle

src/simplify.jl

@@ -39,7 +39,7 @@ function simplify(x;
     end
 
     x = PassThrough(f)(x)
-    simplify_fractions && has_operation(x, /) ?
+    x = simplify_fractions && has_operation(x, /) ?
         SymbolicUtils.simplify_fractions(x) : x
 end
 

src/simplify_rules.jl

@@ -101,6 +101,10 @@ let
         @rule(exp(~x)^(~y) => exp(~x * ~y))
     ]
 
+    TYPE_RULES = [
+        @rule(~x::_isinteger => Int(~x))
+    ]
+
     BOOLEAN_RULES = [
         @rule((true | (~x)) => true)
         @rule(((~x) | true) => true)
@@ -142,6 +146,7 @@ let
                      If(x -> !ispow(x) && is_operation(^)(x),
                         Chain(CANONICALIZE_POW)),
                      If(is_operation(^), Chain(POW_RULES)),
+                     Chain(TYPE_RULES)
                     ] |> RestartedChain
 
         rule_tree

test/rulesets.jl

@@ -57,6 +57,14 @@ end
     @test simplify(Term(zero, [x + 2])) == 0
 end
 
+@testset "Types" begin
+    @syms x
+
+    # should convert to int
+    @test simplify(7.0) === 7
+    @test simplify(7.0*x) === 7*x
+end
+
 @testset "LiteralReal" begin
     @syms x1::LiteralReal x2::LiteralReal
     s = cos(x1 * 3.2) - x2 * 5.8 + x2 * 1.2