From 10ad9a762a8c097ed7d53037bc234db5fcf1cdcd Mon Sep 17 00:00:00 2001
From: Lindsey <lindseyheagy@gmail.com>
Date: Tue, 24 Feb 2015 13:49:33 -0800
Subject: [PATCH] test for Probing diagonal estimator.

---
 SimPEG/Tests/TestUtils.py  |  2 +-
 SimPEG/Tests/test_utils.py | 13 +++++++++++++
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/SimPEG/Tests/TestUtils.py b/SimPEG/Tests/TestUtils.py
index 3cd40991..4292b0b4 100644
--- a/SimPEG/Tests/TestUtils.py
+++ b/SimPEG/Tests/TestUtils.py
@@ -1,7 +1,7 @@
 import numpy as np
 import matplotlib.pyplot as plt
 from numpy.linalg import norm
-from SimPEG.Utils import mkvc, sdiag
+from SimPEG.Utils import mkvc, sdiag, diagEst
 from SimPEG import Utils
 from SimPEG.Mesh import TensorMesh, LogicallyRectMesh, CylMesh
 import numpy as np
diff --git a/SimPEG/Tests/test_utils.py b/SimPEG/Tests/test_utils.py
index 50cc0574..fe9e07c4 100644
--- a/SimPEG/Tests/test_utils.py
+++ b/SimPEG/Tests/test_utils.py
@@ -246,5 +246,18 @@ class TestSequenceFunctions(unittest.TestCase):
         self.assertTrue(np.all(true == listArray))
         self.assertTrue(true.shape == listArray.shape)
 
+class TestDiagEst(unittest.TestCase):
+
+    def setUp(self):
+        self.n = 10
+        self.A = np.random.rand(self.n,self.n)
+        self.Adiag = np.diagonal(self.A)
+
+    def testOnes(self):
+        Adiagtest = diagEst(self.A,self.n,self.n)
+        r = np.abs(Adiagtest-self.Adiag)
+        self.assertTrue(r.dot(r) < TOL)
+
+
 if __name__ == '__main__':
     unittest.main()