Fix python style error

apache · Nov 20, 2019 · 266b9aa · 266b9aa
1 parent 214961b
commit 266b9aa
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 16 deletions.
diff --git a/tests/python/unittest/test_numpy_interoperability.py b/tests/python/unittest/test_numpy_interoperability.py
@@ -1210,24 +1210,24 @@ def get_mat(n):
     A = np.array([[1,2],[3,4],[5,6]])
     vals = (100 * np.arange(5)).astype('l')
     vals_c = (100 * np.array(get_mat(5)) + 1).astype('l')
-    vals_f = _np.array((100 * get_mat(5) + 1), order = 'F', dtype = 'l')
+    vals_f = _np.array((100 * get_mat(5) + 1), order='F', dtype='l')
     vals_f = np.array(vals_f)
 
-    OpArgMngr.add_workload('diagflat', A, k= 2)
-    OpArgMngr.add_workload('diagflat', A, k= 1)
-    OpArgMngr.add_workload('diagflat', A, k= 0)
-    OpArgMngr.add_workload('diagflat', A, k= -1)
-    OpArgMngr.add_workload('diagflat', A, k= -2)
-    OpArgMngr.add_workload('diagflat', A, k= -3)
-    OpArgMngr.add_workload('diagflat', vals, k= 0)
-    OpArgMngr.add_workload('diagflat', vals, k= 2)
-    OpArgMngr.add_workload('diagflat', vals, k= -2)
-    OpArgMngr.add_workload('diagflat', vals_c, k= 0)
-    OpArgMngr.add_workload('diagflat', vals_c, k= 2)
-    OpArgMngr.add_workload('diagflat', vals_c, k= -2)
-    OpArgMngr.add_workload('diagflat', vals_f, k= 0)
-    OpArgMngr.add_workload('diagflat', vals_f, k= 2)
-    OpArgMngr.add_workload('diagflat', vals_f, k= -2)
+    OpArgMngr.add_workload('diagflat', A, k=2)
+    OpArgMngr.add_workload('diagflat', A, k=1)
+    OpArgMngr.add_workload('diagflat', A, k=0)
+    OpArgMngr.add_workload('diagflat', A, k=-1)
+    OpArgMngr.add_workload('diagflat', A, k=-2)
+    OpArgMngr.add_workload('diagflat', A, k=-3)
+    OpArgMngr.add_workload('diagflat', vals, k=0)
+    OpArgMngr.add_workload('diagflat', vals, k=2)
+    OpArgMngr.add_workload('diagflat', vals, k=-2)
+    OpArgMngr.add_workload('diagflat', vals_c, k=0)
+    OpArgMngr.add_workload('diagflat', vals_c, k=2)
+    OpArgMngr.add_workload('diagflat', vals_c, k=-2)
+    OpArgMngr.add_workload('diagflat', vals_f, k=0)
+    OpArgMngr.add_workload('diagflat', vals_f, k=2)
+    OpArgMngr.add_workload('diagflat', vals_f, k=-2)
 
 
 

diff --git a/tests/python/unittest/test_numpy_op.py b/tests/python/unittest/test_numpy_op.py
@@ -4140,6 +4140,7 @@ def dbg(name, data):
                 for (iop, op) in enumerate(grad[0]):
                     assert_almost_equal(grad[0][iop], grad[1][iop], rtol=rtol, atol=atol)
 
+
 @with_seed()
 @use_np
 def test_np_diagflat():