Added match class argument

Author: zewaywong

axelrod/moran.py

@@ -29,6 +29,7 @@ def __init__(
         mutation_method="transition",
         stop_on_fixation=True,
         seed=None,
+        match_class=Match,
     ) -> None:
         """
         An agent based Moran process class. In each round, each player plays a
@@ -106,6 +107,7 @@ def __init__(
         else:
             self.deterministic_cache = DeterministicCache()
         self.turns = turns
+        self.match_class = match_class
         self.prob_end = prob_end
         self.game = game
         self.noise = noise
@@ -376,7 +378,7 @@ def score_all(self) -> List:
         for i, j in self._matchup_indices():
             player1 = self.players[i]
             player2 = self.players[j]
-            match = Match(
+            match = self.match_class(
                 (player1, player2),
                 turns=self.turns,
                 prob_end=self.prob_end,
@@ -574,3 +576,4 @@ def _get_scores_from_cache(self, player_names: Tuple) -> Tuple:
         except KeyError:  # If players are stored in opposite order
             match_scores = self.cached_outcomes[player_names[::-1]].sample()
             return match_scores[::-1]
+            

axelrod/tournament.py

@@ -34,6 +34,7 @@ def __init__(
         edges: List[Tuple] = None,
         match_attributes: dict = None,
         seed: int = None,
+        match_class = Match,
     ) -> None:
         """
         Parameters
@@ -78,6 +79,7 @@ def __init__(
             turns = DEFAULT_TURNS
 
         self.turns = turns
+        self.match_class = match_class
         self.prob_end = prob_end
         self.match_generator = MatchGenerator(
             players=players,
@@ -451,7 +453,7 @@ def _play_matches(self, chunk, build_results=True):
         player2 = self.players[p2_index].clone()
         match_params["players"] = (player1, player2)
         match_params["seed"] = seed
-        match = Match(**match_params)
+        match = self.match_class(**match_params)
         for _ in range(repetitions):
             match.play()