Fix bug for classical nondiagonal noise

src/timeevolution_base.jl

@@ -127,7 +127,7 @@ function integrate_stoch(tspan::Vector{Float64}, df::Function, dg::Function, x0:
         noise_ = noise
     end
     if isa(noise_rate_prototype, Void)
-        if n > 1 || (n > 0 && nc > 0)
+        if n > 1 || nc > 1 || (n > 0 && nc > 0)
             noise_rate_prototype = zeros(Complex128, length(x0), n + nc)
         end
     end
@@ -147,7 +147,7 @@ function integrate_stoch(tspan::Vector{Float64}, df::Function, dg::Function, x0:
                     noise_rate_prototype=noise_rate_prototype)
 
     if isa(alg, Void)
-        if n > 1 || (n > 0 && nc > 0)
+        if n > 1 || nc > 1 || (n > 0 && nc > 0)
             alg_ = StochasticDiffEq.LambaEulerHeun()
         else
             alg_ = StochasticDiffEq.RKMil{:Stratonovich}()

test/test_stochastic_semiclassical.jl

@@ -52,7 +52,11 @@ function fclassical_stoch2(t, psi, u, du)
     du[1,1] = 0.2*u[1]
     du[2,2] = 0.2*u[2]
 end
-
+function fclassical_stoch_ndiag(t, psi, u, du)
+    du[1,1] = 0.2*u[1]
+    du[1,2] = 0.1*u[1]
+    du[2,3] = -0.1u[2]
+end
 
 # Function definitions for master_semiclassical
 function fquantum_master(t, rho, u)
@@ -80,6 +84,10 @@ tout, ψt_sc = stochastic.schroedinger_semiclassical(T_short, ψ_sc, fquantum, f
             fstoch_quantum=fquantum_stoch, fstoch_classical=fclassical_stoch2,
             noise_processes=1,
             noise_prototype_classical=zeros(Complex128, 2,2))
+tout, ψt_sc = stochastic.schroedinger_semiclassical(T_short, ψ_sc, fquantum, fclassical;
+            fstoch_classical=fclassical_stoch_ndiag,
+            noise_prototype_classical=zeros(Complex128, 2, 3))
+
 
 # Semiclassical master
 tout, ρt = stochastic.master_semiclassical(T_short, ρ_sc, fquantum_master, fclassical;