From 266412eb17414c649c72f8a837e1a9aa1ad1f639 Mon Sep 17 00:00:00 2001
From: ChrisRackauckas <chrisrackauckas@gmail.com>
Date: Mon, 12 Jun 2017 09:47:44 -0700
Subject: [PATCH] fix noiseproblem display

---
 src/problems/problem_traits.jl | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/problems/problem_traits.jl b/src/problems/problem_traits.jl
index 082615a2d..556a74593 100644
--- a/src/problems/problem_traits.jl
+++ b/src/problems/problem_traits.jl
@@ -5,6 +5,7 @@ isinplace{uType,iip}(prob::AbstractSteadyStateProblem{uType,iip}) = iip
 isinplace{uType,tType,iip,ND}(prob::AbstractRODEProblem{uType,tType,iip,ND}) = iip
 isinplace{uType,tType,lType,iip}(prob::AbstractDDEProblem{uType,tType,lType,iip}) = iip
 isinplace{uType,duType,tType,iip}(prob::AbstractDAEProblem{uType,duType,tType,iip}) = iip
+isinplace(prob::AbstractNoiseProblem) = isinplace(prob.noise)
 
 ### Displays
 
@@ -15,6 +16,7 @@ end
 
 Base.summary(prob::DEProblem) = string(parameterless_type(prob)," with uType ",typeof(prob.u0)," and tType ",typeof(prob.tspan[1]),". In-place: ",isinplace(prob))
 Base.summary{uType,iip}(prob::AbstractSteadyStateProblem{uType,iip}) = string(parameterless_type(prob)," with uType ",uType)
+Base.summary(prob::AbstractNoiseProblem) = string(parameterless_type(prob)," with WType ",typeof(prob.noise.W[1])," and tType ",typeof(prob.tspan[1]),". In-place: ",isinplace(prob))
 function Base.show(io::IO, A::DEProblem)
   println(io,summary(A))
   print(io,"timespan: ")
@@ -23,6 +25,12 @@ function Base.show(io::IO, A::DEProblem)
   print(io,"u0: ")
   show(io, A.u0)
 end
+function Base.show(io::IO, A::AbstractNoiseProblem)
+  println(io,summary(A))
+  print(io,"timespan: ")
+  show(io,A.tspan)
+  println(io)
+end
 function Base.show(io::IO, A::AbstractDAEProblem)
   println(io,summary(A))
   print(io,"timespan: ")
@@ -47,6 +55,12 @@ function Base.display(io::IO, A::DEProblem)
   print(io,"u0: ")
   display(io, A.u0)
 end
+function Base.display(io::IO, A::AbstractNoiseProblem)
+  println(io,summary(A))
+  print(io,"timespan: ")
+  display(io,A.tspan)
+  println(io)
+end
 function Base.display(io::IO, A::AbstractDAEProblem)
   println(io,summary(A))
   print(io,"timespan: ")