simplify code

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

Author: kylesayrs

tests/test_transform/factory/test_correctness.py

@@ -24,11 +24,9 @@
 
 
 def all_schemes():
-    base = [TransformScheme(type=name) for name in TransformFactory.registered_names()]
-    randomized = [
-        TransformScheme(type=name, randomize=True)
-        for name in TransformFactory.registered_names()
-    ]
+    all_types = TransformFactory.registered_names()
+    base = [TransformScheme(type=type) for type in all_types]
+    randomized = [TransformScheme(type=type, randomize=True) for type in all_types]
     return base + randomized
 
 

tests/test_transform/factory/test_memory.py

@@ -27,11 +27,9 @@
 
 
 def all_schemes():
-    base = [TransformScheme(type=name) for name in TransformFactory.registered_names()]
-    randomized = [
-        TransformScheme(type=name, randomize=True)
-        for name in TransformFactory.registered_names()
-    ]
+    all_types = TransformFactory.registered_names()
+    base = [TransformScheme(type=type) for type in all_types]
+    randomized = [TransformScheme(type=type, randomize=True) for type in all_types]
     return base + randomized