Added singular matrix to test.

cgranade · cgranade · commit c00d2996b748 · 2017-02-09T11:42:08.000+11:00
diff --git a/src/qinfer/tests/test_utils.py b/src/qinfer/tests/test_utils.py
@@ -131,3 +131,13 @@ def test_sqrtm_psd(self):
             np.dot(sqrt_Y, sqrt_Y),
             Y
         )
+
+        # Try again, but with a singular matrix.
+        Y_singular = np.zeros((6, 6))
+        Y_singular[:5, :5] = Y
+        sqrt_Y_singular = sqrtm_psd(Y_singular, est_error=False)
+
+        np.testing.assert_allclose(
+            np.dot(sqrt_Y_singular, sqrt_Y_singular),
+            Y_singular
+        )

Original file line number	Diff line number	Diff line change
`@@ -131,3 +131,13 @@ def test_sqrtm_psd(self):`
`131`	`131`	`np.dot(sqrt_Y, sqrt_Y),`
`132`	`132`	`Y`
`133`	`133`	`)`
	`134`	`+`
	`135`	`+ # Try again, but with a singular matrix.`
	`136`	`+ Y_singular = np.zeros((6, 6))`
	`137`	`+ Y_singular[:5, :5] = Y`
	`138`	`+ sqrt_Y_singular = sqrtm_psd(Y_singular, est_error=False)`
	`139`	`+`
	`140`	`+ np.testing.assert_allclose(`
	`141`	`+ np.dot(sqrt_Y_singular, sqrt_Y_singular),`
	`142`	`+ Y_singular`
	`143`	`+ )`