Fix axis option in Diag test

mikgroup · Oct 30, 2019 · b0889d1 · b0889d1
1 parent 31e013e
commit b0889d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_linop.py b/tests/test_linop.py
@@ -175,7 +175,7 @@ def test_Diag(self):
         I = linop.Identity(shape)
         x = util.randn([5, 6])
 
-        A = linop.Diag([I, I], axis=1)
+        A = linop.Diag([I, I], iaxis=1, oaxis=1)
         npt.assert_allclose(A(x), x)
         self.check_linop_linear(A)
         self.check_linop_adjoint(A)