Create and use axl_filename

Author: gaffney2010

axelrod/load_data_.py

@@ -1,8 +1,19 @@
-from typing import Dict, List, Tuple
+from typing import Dict, List, Text, Tuple
 
+import os
 import pkg_resources
 
 
+def axl_filename(axl_path: Text) -> Text:
+    """Get the path to Axelrod/<axl_path> from the working directory."""
+    # Working directory
+    dirname = os.path.dirname(__file__)
+
+    # We go up a dir because this code is located in Axelrod/axelrod and
+    # axl_path is from the top-level Axelrod dir.
+    return os.path.join(dirname, "..", axl_path)
+
+
 def load_file(filename: str, directory: str) -> List[List[str]]:
     """Loads a data file stored in the Axelrod library's data subdirectory,
     likely for parameters for a strategy."""

axelrod/plot.py

@@ -8,6 +8,7 @@
 from numpy import arange, median, nan_to_num
 
 from .result_set import ResultSet
+from .load_data_ import axl_filename
 
 titleType = List[str]
 namesType = List[str]
@@ -323,7 +324,7 @@ def save_all_plots(
 
         for method, name in plots:
             f = getattr(self, method)(title="{} - {}".format(title_prefix, name))
-            f.savefig("{}_{}.{}".format(prefix, method, filetype))
+            f.savefig(axl_filename("{}_{}.{}".format(prefix, method, filetype)))
             plt.close(f)
 
             if progress_bar:

axelrod/tests/integration/test_tournament.py

@@ -4,6 +4,7 @@
 from hypothesis import given, settings
 
 import axelrod
+from axelrod.load_data_ import axl_filename
 from axelrod.strategy_transformers import FinalTransformer
 from axelrod.tests.property import tournaments
 
@@ -45,7 +46,7 @@ def setUpClass(cls):
     def test_big_tournaments(self, tournament):
         """A test to check that tournament runs with a sample of non-cheating
         strategies."""
-        filename = "test_outputs/test_tournament.csv"
+        filename = axl_filename("test_outputs/test_tournament.csv")
         self.assertIsNone(
             tournament.play(progress_bar=False, filename=filename, build_results=False)
         )
@@ -90,7 +91,7 @@ def test_repeat_tournament_deterministic(self):
                 turns=2,
                 repetitions=2,
             )
-            files.append("test_outputs/stochastic_tournament_{}.csv".format(_))
+            files.append(axl_filename("test_outputs/stochastic_tournament_{}.csv".format(_)))
             tournament.play(progress_bar=False, filename=files[-1], build_results=False)
         self.assertTrue(filecmp.cmp(files[0], files[1]))
 
@@ -113,7 +114,7 @@ def test_repeat_tournament_stochastic(self):
                 turns=2,
                 repetitions=2,
             )
-            files.append("test_outputs/stochastic_tournament_{}.csv".format(_))
+            files.append(axl_filename("test_outputs/stochastic_tournament_{}.csv".format(_)))
             tournament.play(progress_bar=False, filename=files[-1], build_results=False)
         self.assertTrue(filecmp.cmp(files[0], files[1]))
 

axelrod/tests/unit/test_deterministic_cache.py

@@ -3,6 +3,7 @@
 import unittest
 
 from axelrod import Action, Defector, DeterministicCache, Random, TitForTat
+from axelrod.load_data_ import axl_filename
 
 C, D = Action.C, Action.D
 
@@ -12,8 +13,8 @@ class TestDeterministicCache(unittest.TestCase):
     def setUpClass(cls):
         cls.test_key = (TitForTat(), Defector())
         cls.test_value = [(C, D), (D, D), (D, D)]
-        cls.test_save_file = "test_cache_save.txt"
-        cls.test_load_file = "test_cache_load.txt"
+        cls.test_save_file = axl_filename("test_outputs/test_cache_save.txt")
+        cls.test_load_file = axl_filename("test_outputs/test_cache_load.txt")
         test_data_to_pickle = {("Tit For Tat", "Defector"): [(C, D), (D, D), (D, D)]}
         cls.test_pickle = pickle.dumps(test_data_to_pickle)
 
@@ -92,7 +93,7 @@ def test_load(self):
         self.assertEqual(self.cache[self.test_key], self.test_value)
 
     def test_load_error_for_inccorect_format(self):
-        filename = "test_outputs/test.cache"
+        filename = axl_filename("test_outputs/test.cache")
         with open(filename, "wb") as io:
             pickle.dump(range(5), io)
 

axelrod/tests/unit/test_fingerprint.py

@@ -9,6 +9,7 @@
 
 import axelrod as axl
 from axelrod.fingerprint import AshlockFingerprint, Point, TransitiveFingerprint
+from axelrod.load_data_ import axl_filename
 from axelrod.strategy_transformers import DualTransformer, JossAnnTransformer
 from axelrod.tests.property import strategy_lists
 
@@ -195,7 +196,7 @@ def test_temp_file_creation(self):
 
         RecordedMksTemp.reset_record()
         af = AshlockFingerprint(axl.TitForTat)
-        filename = "test_outputs/test_fingerprint.csv"
+        filename = axl_filename("test_outputs/test_fingerprint.csv")
 
         self.assertEqual(RecordedMksTemp.record, [])
 
@@ -211,7 +212,7 @@ def test_temp_file_creation(self):
         self.assertFalse(os.path.isfile(filename))
 
     def test_fingerprint_with_filename(self):
-        filename = "test_outputs/test_fingerprint.csv"
+        filename = axl_filename("test_outputs/test_fingerprint.csv")
         af = AshlockFingerprint(axl.TitForTat)
         af.fingerprint(
             turns=1, repetitions=1, step=0.5, progress_bar=False, filename=filename
@@ -425,7 +426,7 @@ def test_init_with_not_default_number(self):
         )
 
     def test_fingerprint_with_filename(self):
-        filename = "test_outputs/test_fingerprint.csv"
+        filename = axl_filename("test_outputs/test_fingerprint.csv")
         strategy = axl.TitForTat()
         tf = TransitiveFingerprint(strategy)
         tf.fingerprint(turns=1, repetitions=1, progress_bar=False, filename=filename)
@@ -437,7 +438,9 @@ def test_serial_fingerprint(self):
         strategy = axl.TitForTat()
         tf = TransitiveFingerprint(strategy)
         tf.fingerprint(
-            repetitions=1, progress_bar=False, filename="test_outputs/tran_fin.csv"
+            repetitions=1,
+            progress_bar=False,
+            filename=axl_filename("test_outputs/tran_fin.csv"),
         )
         self.assertEqual(tf.data.shape, (50, 50))
 
@@ -450,7 +453,7 @@ def test_parallel_fingerprint(self):
 
     def test_analyse_cooperation_ratio(self):
         tf = TransitiveFingerprint(axl.TitForTat)
-        filename = "test_outputs/test_fingerprint.csv"
+        filename = axl_filename("test_outputs/test_fingerprint.csv")
         with open(filename, "w") as f:
             f.write(
                 """Interaction index,Player index,Opponent index,Repetition,Player name,Opponent name,Actions

axelrod/tests/unit/test_plot.py

@@ -2,6 +2,7 @@
 import unittest
 
 import axelrod
+from axelrod.load_data_ import axl_filename
 import matplotlib
 import matplotlib.pyplot as plt
 from numpy import mean
@@ -10,7 +11,7 @@
 class TestPlot(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
-        cls.filename = "test_outputs/test_results.csv"
+        cls.filename = axl_filename("test_outputs/test_results.csv")
 
         cls.players = [axelrod.Alternator(), axelrod.TitForTat(), axelrod.Defector()]
         cls.repetitions = 3

axelrod/tests/unit/test_resultset.py

@@ -5,6 +5,7 @@
 import axelrod
 import axelrod.interaction_utils as iu
 import pandas as pd
+from axelrod.load_data_ import axl_filename
 from axelrod.result_set import create_counter_dict
 from axelrod.tests.property import prob_end_tournaments, tournaments
 from numpy import mean, nanmedian, std
@@ -19,7 +20,7 @@ class TestResultSet(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
 
-        cls.filename = "test_outputs/test_results.csv"
+        cls.filename = axl_filename("test_outputs/test_results.csv")
 
         cls.players = [axelrod.Alternator(), axelrod.TitForTat(), axelrod.Defector()]
         cls.repetitions = 3
@@ -647,7 +648,7 @@ class TestResultSetSpatialStructure(TestResultSet):
     @classmethod
     def setUpClass(cls):
 
-        cls.filename = "test_outputs/test_results_spatial.csv"
+        cls.filename = axl_filename("test_outputs/test_results_spatial.csv")
         cls.players = [axelrod.Alternator(), axelrod.TitForTat(), axelrod.Defector()]
         cls.turns = 5
         cls.edges = [(0, 1), (0, 2)]

axelrod/tests/unit/test_tournament.py

@@ -12,6 +12,7 @@
 from unittest.mock import MagicMock, patch
 
 import axelrod
+from axelrod.load_data_ import axl_filename
 import numpy as np
 import pandas as pd
 from axelrod.tests.property import (
@@ -88,7 +89,7 @@ def setUpClass(cls):
             [200, 200, 1, 200, 200],
         ]
 
-        cls.filename = "test_outputs/test_tournament.csv"
+        cls.filename = axl_filename("test_outputs/test_tournament.csv")
 
     def setUp(self):
         self.test_tournament = axelrod.Tournament(
@@ -733,7 +734,9 @@ def test_write_to_csv_with_results(self):
         )
         tournament.play(filename=self.filename, progress_bar=False)
         df = pd.read_csv(self.filename)
-        expected_df = pd.read_csv("test_outputs/expected_test_tournament.csv")
+        expected_df = pd.read_csv(
+            axl_filename("test_outputs/expected_test_tournament.csv")
+        )
         self.assertTrue(df.equals(expected_df))
 
     def test_write_to_csv_without_results(self):
@@ -747,7 +750,7 @@ def test_write_to_csv_without_results(self):
         tournament.play(filename=self.filename, progress_bar=False, build_results=False)
         df = pd.read_csv(self.filename)
         expected_df = pd.read_csv(
-            "test_outputs/expected_test_tournament_no_results.csv"
+            axl_filename("test_outputs/expected_test_tournament_no_results.csv")
         )
         self.assertTrue(df.equals(expected_df))