up

Author: vyudu

src/utils.jl

@@ -158,7 +158,7 @@ end
 function check_var_types(sys_type::Type{T}, dvs) where T <: AbstractSystem
     any(u -> !(symtype(u) <: Number), dvs) && error("The type of unknown variables must be a numeric type.")
     if sys_type == ODESystem || sys_type == SDESystem || sys_type == PDESystem
-        any(u -> !(symtype(u) <: Float), dvs) && error("The type of unknown variables for an SDESystem, PDESystem, or ODESystem must be a float.")
+        any(u -> !(symtype(u) <: AbstractFloat), dvs) && error("The type of unknown variables for an SDESystem, PDESystem, or ODESystem must be a float.")
     end
     nothing
 end

test/odesystem.jl

@@ -1552,3 +1552,10 @@ end
     expected_tstops = unique!(sort!(vcat(0.0:0.075:10.0, 0.1, 0.2, 0.65, 0.35, 0.45)))
     @test all(x -> any(isapprox(x, atol = 1e-6), sol2.t), expected_tstops)
 end
+
+let
+    @parameters p d
+    @variables X(t)::Int64
+    eq = D(X) ~ p - d*X
+    @test_throws Exception @mtkbuild osys = ODESystem([eq], t)
+end

test/sdesystem.jl

@@ -868,3 +868,12 @@ end
     @test length(ModelingToolkit.get_noiseeqs(sys)) == 1
     @test length(observed(sys)) == 1
 end
+
+# Test validating types of states
+let
+    @parameters p d
+    @variables X(t)::Int64
+    @brownian z
+    eq2 = D(X) ~ p - d*X + z
+    @test_throws Exception @mtkbuild ssys = System([eq2], t)
+end