style fix in one test_pickling test

dmanc · Aug 3, 2017 · 91c28e9 · 91c28e9
1 parent 0831737
commit 91c28e9
Showing 1 changed file with 7 additions and 11 deletions.
diff --git a/axelrod/tests/unit/test_pickling.py b/axelrod/tests/unit/test_pickling.py
@@ -247,17 +247,13 @@ def test_dual_transformer_regression_test(self):
         player = InterspersedDualTransformersNamePrefixPresent()
         self.assert_orignal_equals_pickled(player)
 
-        interspersed_dual_transformers = (
-            st.TrackHistoryTransformer()(
-                st.DualTransformer()(
-                    st.InitialTransformer((C, D))(
-                        st.DualTransformer()(
-                            axl.WinStayLoseShift
-                        )
-                    )
-                )
-            )()
-        )
+        player_class = axl.WinStayLoseShift
+        player_class = st.DualTransformer()(player_class)
+        player_class = st.InitialTransformer((C, D))(player_class)
+        player_class = st.DualTransformer()(player_class)
+        player_class = st.TrackHistoryTransformer()(player_class)
+
+        interspersed_dual_transformers = player_class()
 
         self.assert_orignal_equals_pickled(interspersed_dual_transformers)