add some funcs to common

qiskit-community · Nov 2, 2023 · 5d99e54 · 5d99e54
1 parent 9e9529c
commit 5d99e54
Showing 1 changed file with 16 additions and 16 deletions.
diff --git a/test/dynamics/common.py b/test/dynamics/common.py
@@ -89,13 +89,13 @@ def asarray(self, a):
         """Array generation method."""
         return np.array(a)
 
-    def zeros(self, shape, dtype= None):
+    def zeros(self, shape, dtype=None):
         """Array filled with zeros generation method."""
-        return np.zeros(shape,dtype)
+        return np.zeros(shape, dtype)
 
-    def ones(self, shape, dtype= None):
+    def ones(self, shape, dtype=None):
         """Array filled with ones generation method."""
-        return np.ones(shape,dtype)
+        return np.ones(shape, dtype)
 
     def assertArrayType(self, a):
         """Assert the correct array type."""
@@ -125,13 +125,13 @@ def asarray(self, a):
         """Array generation method."""
         return jnp.array(a)
 
-    def zeros(self, shape, dtype= None):
+    def zeros(self, shape, dtype=None):
         """Array filled with zeros generation method."""
-        return jnp.zeros(shape,dtype)
+        return jnp.zeros(shape, dtype)
 
-    def ones(self, shape, dtype= None):
+    def ones(self, shape, dtype=None):
         """Array filled with ones generation method."""
-        return jnp.ones(shape,dtype)
+        return jnp.ones(shape, dtype)
 
     def assertArrayType(self, a):
         """Assert the correct array type."""
@@ -150,13 +150,13 @@ def asarray(self, a):
         """Array generation method."""
         return Array(a)
 
-    def zeros(self, shape, dtype= None):
+    def zeros(self, shape, dtype=None):
         """Array filled with zeros generation method."""
-        return Array(np.zeros(shape,dtype))
+        return Array(np.zeros(shape, dtype))
 
-    def ones(self, shape, dtype= None):
+    def ones(self, shape, dtype=None):
         """Array filled with ones generation method."""
-        return Array(np.ones(shape,dtype))
+        return Array(np.ones(shape, dtype))
 
     def assertArrayType(self, a):
         """Assert the correct array type."""
@@ -193,13 +193,13 @@ def asarray(self, a):
         """Array generation method."""
         return Array(a)
 
-    def zeros(self, shape, dtype= None):
+    def zeros(self, shape, dtype=None):
         """Array filled with zeros generation method."""
-        return Array(jnp.zeros(shape,dtype))
+        return Array(jnp.zeros(shape, dtype))
 
-    def ones(self, shape, dtype= None):
+    def ones(self, shape, dtype=None):
         """Array filled with ones generation method."""
-        return Array(jnp.ones(shape,dtype))
+        return Array(jnp.ones(shape, dtype))
 
     def assertArrayType(self, a):
         """Assert the correct array type."""