check for names

Author: eric-s-s

axelrod/strategy_transformers.py

@@ -143,6 +143,15 @@ def __repr__(self):
 
             # Define a new class and wrap the strategy method
             # Dynamically create the new class
+            def reducer(self_):
+                class_module = __import__(self_.__module__)
+                if self_.__class__.__name__ in dir(class_module):
+                    return self_.__class__, (), self_.__dict__
+                else:
+                    print('oops {}  {}'.format(
+                        self_.__class__.__name__, self_.original_class.__name__
+                    ))
+
             new_class = type(
                 new_class_name, (PlayerClass,),
                 {
@@ -153,6 +162,7 @@ def __repr__(self):
                     "__module__": PlayerClass.__module__,
                     "classifier": classifier,
                     "__doc__": PlayerClass.__doc__,
+                    "__reduce__": reducer
                 })
             return new_class
     return Decorator

axelrod/test_pickling.py

@@ -37,6 +37,22 @@ def test_final_transformer_called(self):
         results = match.play()
         self.assertEqual(results, [(C, C), (C, C), (D, D)])
 
+    def test_nice_transformer_called(self):
+        player = axl.NMWEDeterministic()
+        copy = pickle.loads(pickle.dumps(player))
+        opponent_1 = axl.CyclerCCCDCD()
+        opponent_2 = axl.CyclerCCCDCD()
+        axl.seed(0)
+        match_1 = axl.Match((player, opponent_1), turns=100)
+        result_1 = match_1.play()
+
+        axl.seed(0)
+        match_2 = axl.Match((copy, opponent_2), turns=100)
+        result_2 = match_2.play()
+
+
+        self.assertEqual(result_1, result_2)
+
     def test_all(self):
         for s in axl.strategies:
             player = s()